Update app.py

app.py

@@ -118,29 +118,6 @@ pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
 
 MAX_SEED = np.iinfo(np.int32).max
 
-# --- Helper Functions ---
-def update_dimensions_on_upload(image):
-    if image is None:
-        return 1024, 1024
-    
-    original_width, original_height = image.size
-    
-    # Cap max dimension to 1024 while preserving aspect ratio
-    if original_width > original_height:
-        new_width = 1024
-        aspect_ratio = original_height / original_width
-        new_height = int(new_width * aspect_ratio)
-    else:
-        new_height = 1024
-        aspect_ratio = original_width / original_height
-        new_width = int(new_height * aspect_ratio)
-        
-    # Ensure dimensions are multiples of 8 for model compatibility
-    new_width = (new_width // 8) * 8
-    new_height = (new_height // 8) * 8
-    
-    return new_width, new_height
-
 # --- Main Inference Function ---
 @spaces.GPU
 def infer(
@@ -190,6 +167,29 @@ def infer(
 
     return result, seed, gr.Button(visible=True)
 
+# --- Helper Functions ---
+def update_dimensions_on_upload(image):
+    if image is None:
+        return 1024, 1024
+    
+    original_width, original_height = image.size
+    
+    # Cap max dimension to 1024 while preserving aspect ratio
+    if original_width > original_height:
+        new_width = 1024
+        aspect_ratio = original_height / original_width
+        new_height = int(new_width * aspect_ratio)
+    else:
+        new_height = 1024
+        aspect_ratio = original_width / original_height
+        new_width = int(new_height * aspect_ratio)
+        
+    # Ensure dimensions are multiples of 8 for model compatibility
+    new_width = (new_width // 8) * 8
+    new_height = (new_height // 8) * 8
+    
+    return new_width, new_height
+
 # Wrapper for examples to handle file paths
 @spaces.GPU
 def infer_example(input_image_path, prompt, lora_adapter):
@@ -277,4 +277,4 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
         outputs=[width, height]
     )
 
-demo.launch(mcp_server=True, ssr_mode=False, show_error=True)
+demo.launch(ssr_mode=False)