Spaces:
Build error
Build error
Update gradio_app.py
Browse files- gradio_app.py +1 -1
gradio_app.py
CHANGED
|
@@ -15,7 +15,7 @@ from huggingface_hub import snapshot_download
|
|
| 15 |
weights_dir = './allegro_weights'
|
| 16 |
os.makedirs(weights_dir, exist_ok=True)
|
| 17 |
|
| 18 |
-
is_shared_ui = True if "fffiloni/allegro-
|
| 19 |
is_gpu_associated = torch.cuda.is_available()
|
| 20 |
|
| 21 |
if not is_shared_ui:
|
|
|
|
| 15 |
weights_dir = './allegro_weights'
|
| 16 |
os.makedirs(weights_dir, exist_ok=True)
|
| 17 |
|
| 18 |
+
is_shared_ui = True if "fffiloni/allegro-text2video" in os.environ['SPACE_ID'] else False
|
| 19 |
is_gpu_associated = torch.cuda.is_available()
|
| 20 |
|
| 21 |
if not is_shared_ui:
|