Update app.py

app.py

@@ -222,9 +222,7 @@ with gr.Blocks(delete_cache=(3600, 3600)) as demo:
                 columns=2,
                 rows=1,
                 object_fit="scale-down",
-                show_download_button=True,
-                type="pil",
-                show_share_button=False
+                show_download_button=True
             )
 
     # --- Event Handling ---
@@ -243,6 +241,10 @@ with gr.Blocks(delete_cache=(3600, 3600)) as demo:
         fn=upscale_with_all_models,
         inputs=[input_image],
         outputs=[all_models_gallery],
+    ).then(
+        lambda x: x,
+        inputs=[all_models_gallery],
+        outputs=[all_models_gallery]
     )
 
 # --- Pre-load the default model for a faster first-time user experience ---