add directed ncut (test)

app.py

@@ -204,9 +204,9 @@ def compute_ncut_directed(
     make_symmetric=False,
     progess_start=0.4,
 ):        
-    print("Using directed_ncut")
-    print("features_1.shape", features_1.shape)
-    print("features_2.shape", features_2.shape)
+    # print("Using directed_ncut")
+    # print("features_1.shape", features_1.shape)
+    # print("features_2.shape", features_2.shape)
     from directed_ncut import nystrom_ncut
     progress = gr.Progress()
     logging_str = ""
@@ -977,26 +977,26 @@ def ncut_run(
 
 def _ncut_run(*args, **kwargs):
     n_ret = kwargs.pop("n_ret", 1)
-    # try:
-    #     if torch.cuda.is_available():
-    #         torch.cuda.empty_cache()
+    try:
+        if torch.cuda.is_available():
+            torch.cuda.empty_cache()
             
-    #     ret = ncut_run(*args, **kwargs)
+        ret = ncut_run(*args, **kwargs)
         
-    #     if torch.cuda.is_available():
-    #         torch.cuda.empty_cache()
+        if torch.cuda.is_available():
+            torch.cuda.empty_cache()
         
-    #     ret = list(ret)[:n_ret] + [ret[-1]]
-    #     return ret
-    # except Exception as e:
-    #     gr.Error(str(e))
-    #     if torch.cuda.is_available():
-    #         torch.cuda.empty_cache()
-    #     return *(None for _ in range(n_ret)), "Error: " + str(e)
+        ret = list(ret)[:n_ret] + [ret[-1]]
+        return ret
+    except Exception as e:
+        gr.Error(str(e))
+        if torch.cuda.is_available():
+            torch.cuda.empty_cache()
+        return *(None for _ in range(n_ret)), "Error: " + str(e)
 
-    ret = ncut_run(*args, **kwargs)
-    ret = list(ret)[:n_ret] + [ret[-1]]
-    return ret
+    # ret = ncut_run(*args, **kwargs)
+    # ret = list(ret)[:n_ret] + [ret[-1]]
+    # return ret
 
 if USE_HUGGINGFACE_ZEROGPU:
     @spaces.GPU(duration=30)
@@ -1213,7 +1213,7 @@ def run_fn(
     advanced=False,
     directed=False,
 ):
-    print(node_type2, head_index_text, make_symmetric)
+    # print(node_type2, head_index_text, make_symmetric)
     progress=gr.Progress()
     progress(0, desc="Starting")