Update app.py

app.py

@@ -488,6 +488,9 @@ def batch_realesrgan(
     yield gallery, str(zip_path), details, render_progress(100, "Complete")
 
 
+# ────────────────────────────────────────────────────────
+# UI
+# ────────────────────────────────────────────────────────
 # ────────────────────────────────────────────────────────
 # UI
 # ────────────────────────────────────────────────────────
@@ -541,7 +544,7 @@ def main():
         )
         reset_btn.click(fn=reset, inputs=[], outputs=[output_image, input_image])
 
-                # --- Batch Upscale (multi-image) ---
+        # --- Batch Upscale (multi-image) ---
         gr.Markdown("### Batch Upscale")
         with gr.Accordion("Batch options", open=True):
             with gr.Row():
@@ -549,6 +552,7 @@ def main():
                     label="Upload multiple images (PNG/JPG/JPEG)",
                     type="filepath",
                     file_types=[".png", ".jpg", ".jpeg"],
+                    # file_count="multiple"  # optional: explicit in some versions
                 )
             with gr.Row():
                 batch_tile = gr.Number(label="Tile size (0/auto → 256)", value=256, precision=0)
@@ -568,17 +572,12 @@ def main():
             inputs=[batch_files, model_name, denoise_strength, face_enhance, outscale, batch_tile, batch_size],
             outputs=[batch_gallery, batch_zip, batch_details, batch_prog],
         )
-        gr.Markdown("")  # spacer
 
-    # Disable SSR (ZeroGPU + Gradio logs suggested turning this off)
-    demo.launch(ssr_mode=False)  # set share=True for a public link
+        gr.Markdown("")  # spacer
 
+    return demo
 
-def main():
-    with gr.Blocks(title="Real-ESRGAN Gradio Demo", theme="ParityError/Interstellar") as demo:
-        # ... your current UI (plus batch section) ...
-        return demo
 
 if __name__ == "__main__":
     demo = main()
-    demo.queue().launch(ssr_mode=False)
+    demo.queue().launch(ssr_mode=False)  # set share=True if you want a public link