cpu for ncut

app.py

@@ -304,7 +304,7 @@ def ncut_run(
                 affinity_focal_gamma=affinity_focal_gamma,
                 knn_ncut=1,
                 knn_tsne=10,
-                num_sample_tsne=500,
+                num_sample_tsne=300,
                 embedding_method=embedding_method,
                 perplexity=perplexity,
                 n_neighbors=n_neighbors,
@@ -381,6 +381,9 @@ if USE_HUGGINGFACE_ZEROGPU:
     @spaces.GPU(duration=120)
     def super_duper_long_run(*args, **kwargs):
         return _ncut_run(*args, **kwargs)
+    
+    def cpu_run(*args, **kwargs):
+        return _ncut_run(*args, **kwargs)
 
 if not USE_HUGGINGFACE_ZEROGPU:
     def quick_run(*args, **kwargs):
@@ -394,6 +397,9 @@ if not USE_HUGGINGFACE_ZEROGPU:
 
     def super_duper_long_run(*args, **kwargs):
         return _ncut_run(*args, **kwargs)
+    
+    def cpu_run(*args, **kwargs):
+        return _ncut_run(*args, **kwargs)
 
 def extract_video_frames(video_path, max_frames=100):
     from decord import VideoReader
@@ -524,6 +530,10 @@ def run_fn(
         "video_output": video_output,
     }
     # print(kwargs)
+    
+    if old_school_ncut:
+        cpu_run(model, images, **kwargs)
+    
     num_images = len(images)
     if num_images >= 100:
         return super_duper_long_run(model, images, **kwargs)