Spaces:
Configuration error
Configuration error
Commit
•
5fc2e96
1
Parent(s):
d1c0e9f
Update the `is_shared_ui` for not needing an environment variable
Browse files
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 |
-
|
|
|
|
|
|
|
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 = '''
|