fix space

app.py

@@ -1,13 +1,13 @@
 # Author: Huzheng Yang
 # %%
-HUGGINGFACE_SPACE = True
+USE_HUGGINGFACE_SPACE = True
 BATCH_SIZE = 4
 
-if HUGGINGFACE_SPACE:  # huggingface ZeroGPU, dynamic GPU allocation 
+if USE_HUGGINGFACE_SPACE:  # huggingface ZeroGPU, dynamic GPU allocation 
     try:
         import spaces
     except ImportError:
-        HUGGINGFACE_SPACE = False  # run on local machine
+        USE_HUGGINGFACE_SPACE = False  # run on local machine
         BATCH_SIZE = 1
 
 import os
@@ -309,7 +309,7 @@ def _ncut_run(*args, **kwargs):
             torch.cuda.empty_cache()
         return [], "Error: " + str(e)
 
-if HUGGINGFACE_SPACE:
+if USE_HUGGINGFACE_SPACE:
     @spaces.GPU(duration=20)
     def quick_run(*args, **kwargs):
         return _ncut_run(*args, **kwargs)
@@ -326,7 +326,7 @@ if HUGGINGFACE_SPACE:
     def super_duper_long_run(*args, **kwargs):
         return _ncut_run(*args, **kwargs)
 
-if not HUGGINGFACE_SPACE:
+if not USE_HUGGINGFACE_SPACE:
     def quick_run(*args, **kwargs):
         return _ncut_run(*args, **kwargs)
 
@@ -905,7 +905,7 @@ with demo:
         
 
 
-if HUGGINGFACE_SPACE:
+if USE_HUGGINGFACE_SPACE:
     download_all_models()
     download_all_datasets()
     demo.launch()

requirements.txt

@@ -1,4 +1,4 @@
-torch==2.3.1
+torch
 torchvision 
 ncut-pytorch
 opencv-python