fix eigvecs

app.py

@@ -169,6 +169,7 @@ def compute_ncut(
     
     if only_eigvecs:
         eigvecs = eigvecs.to("cpu").reshape(features.shape[:-1] + (num_eig,))
+        eigvecs = eigvecs.detach().numpy()
         return None, logging_str, eigvecs
     
     start = time.time()
@@ -3443,7 +3444,7 @@ with demo:
                 
     
     with gr.Tab('PlayGround (test)', visible=False) as test_playground_tab:
-        eigvecs = gr.State(torch.tensor([]))
+        eigvecs = gr.State(np.array([]))
         with gr.Row():
             with gr.Column(scale=5, min_width=200):
                 gr.Markdown("### Step 1: Load Images and Run NCUT")
@@ -3605,6 +3606,7 @@ with demo:
         @torch.no_grad()
         def run_heatmap(images, eigvecs, image1_slider, prompt_image1, n_eig, distance_slider, flat_idx=None, overlay_image=True):
             gr.Info(f"current number of eigenvectors: {n_eig}")
+            eigvecs = torch.tensor(eigvecs)
             images = [image[0] for image in images]
             if isinstance(images[0], str):
                 images = [Image.open(image[0]).convert("RGB").resize((256, 256)) for image in images]
@@ -3711,7 +3713,7 @@ with demo:
             )
             
     with gr.Tab('PlayGround', visible=True) as test_playground_tab2:
-        eigvecs = gr.State(torch.tensor([]))
+        eigvecs = gr.State(np.array([]))
         with gr.Row():
             with gr.Column(scale=5, min_width=200):
                 gr.Markdown("### Step 1: Load Images")
@@ -3856,6 +3858,7 @@ with demo:
         @torch.no_grad()
         def run_heatmap(images, eigvecs, image1_slider, prompt_image1, n_eig, distance_slider, flat_idx=None, overlay_image=True):
             gr.Info(f"current number of eigenvectors: {n_eig}", 2)
+            eigvecs = torch.tensor(eigvecs)
             images = [image[0] for image in images]
             if isinstance(images[0], str):
                 images = [Image.open(image[0]).convert("RGB").resize((256, 256)) for image in images]