Update

app.py

@@ -162,7 +162,7 @@ def segment(inp: Image):
     # Arrange eigenvectors into grid
     cmap = get_cmap('viridis')
     output_images = []
-    eigenvectors_upscaled = []
+    # eigenvectors_upscaled = []
     for i in range(1, K + 1):
         eigenvector = eigenvectors[i].reshape(1, 1, H_patch, W_patch)  # .reshape(1, 1, H_pad, W_pad)
         eigenvector: torch.Tensor = F.interpolate(eigenvector, size=(H_pad, W_pad), mode='bilinear', align_corners=False)  # slightly off, but for visualizations this is okay
@@ -172,7 +172,7 @@ def segment(inp: Image):
         eigenvector_vis = Image.open(buffer).convert('RGB')
         # eigenvector_vis = TF.to_tensor(eigenvector_vis).unsqueeze(0)
         eigenvector_vis = np.array(eigenvector_vis)
-        eigenvectors_upscaled.append(eigenvector)
+        # eigenvectors_upscaled.append(eigenvector)
         output_images.append(eigenvector_vis)
     # output_images = torch.cat(output_images, dim=0)
     # output_images = make_grid(output_images, nrow=8, pad_value=1)