update deploy

app.py

@@ -2,9 +2,15 @@
 # %%
 import os
 USE_HUGGINGFACE_ZEROGPU = os.getenv("USE_HUGGINGFACE_ZEROGPU", "False").lower() in ["true", "1", "yes"]
+DOWNLOAD_ALL_MODELS_DATASETS = os.getenv("DOWNLOAD_ALL_MODELS_DATASETS", "False").lower() in ["true", "1", "yes"]
 #%%
 if USE_HUGGINGFACE_ZEROGPU:  # huggingface ZeroGPU, dynamic GPU allocation 
-    import spaces
+    try:
+        import spaces
+    except:
+        USE_HUGGINGFACE_ZEROGPU = False
+        
+if USE_HUGGINGFACE_ZEROGPU:
     BATCH_SIZE = 4
 else:  # run on local machine
     BATCH_SIZE = 1
@@ -900,15 +906,14 @@ with demo:
         
 
 
-if USE_HUGGINGFACE_ZEROGPU:
+if DOWNLOAD_ALL_MODELS_DATASETS:
     from ncut_pytorch.backbone import download_all_models
     threading.Thread(target=download_all_models).start()
     from ncut_pytorch.backbone_text import download_all_models
     threading.Thread(target=download_all_models).start()
     
     threading.Thread(target=download_all_datasets).start()
-    demo.launch()
-else:
-    demo.launch(share=True)
+    
+demo.launch(share=True)
 
 # %%