multimodalart HF staff commited on
Commit
5fc2e96
1 Parent(s): d1c0e9f

Update the `is_shared_ui` for not needing an environment variable

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -18,7 +18,10 @@ from huggingface_hub import snapshot_download
18
 
19
 
20
  is_spaces = True if "SPACE_ID" in os.environ else False
21
- is_shared_ui = True if "IS_SHARED_UI" in os.environ else False
 
 
 
22
  is_gpu_associated = torch.cuda.is_available()
23
 
24
  css = '''
 
18
 
19
 
20
  is_spaces = True if "SPACE_ID" in os.environ else False
21
+ if(is_spaces):
22
+ is_shared_ui = True if "BAAI/dreambooth-altdiffusion" in os.environ['SPACE_ID'] else False
23
+ else:
24
+ is_shared_ui = False
25
  is_gpu_associated = torch.cuda.is_available()
26
 
27
  css = '''