update UI

app.py

@@ -119,10 +119,9 @@ def dont_use_too_much_green(image_rgb):
     return image_rgb
 
 
-def to_pil_images(images):
+def to_pil_images(images, target_size=256):
     size = images[0].shape[1]
-    target = 256
-    multiplier = target // size
+    multiplier = target_size // size
     res = int(size * multiplier)
     return [
         Image.fromarray((image * 255).cpu().numpy().astype(np.uint8)).resize((res, res), Image.Resampling.NEAREST)
@@ -337,20 +336,11 @@ def ncut_run(
             galleries = []
             for i_node in range(rgb.shape[1]):
                 _rgb = rgb[:, i_node]
-                galleries.append(to_pil_images(_rgb))
+                galleries.append(to_pil_images(_rgb, target_size=56))
             return *galleries, logging_str
         
         rgb = dont_use_too_much_green(rgb)
         
-    if "AlignedThreeModelAttnNodes" == model_name:
-        # dirty patch for the alignedcut paper
-        print("AlignedThreeModelAttnNodes")
-        galleries = []
-        for i_node in range(rgb.shape[1]):
-            _rgb = rgb[:, i_node]
-            print(_rgb.shape)
-            galleries.append(to_pil_images(_rgb))
-        return *galleries, logging_str
     
     if video_output:
         video_path = get_random_path()