Spaces:
Running
Running
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:
|