Update app.py

app.py

@@ -188,7 +188,7 @@ def generate_image_handler(x, negative_prompt, fine_edge, fix_perspective, grow_
         # 2. Instantiate a client specifically for this request with the forwarded headers.
         # This ensures the backend sees the 'x-zerogpu-token' of the user, not the server.
         # gradio_client caches schemas, so re-init is relatively cheap but necessary for headers.
-        client = Client(BACKEND_URL)
+        client = Client(BACKEND_URL, headers=forward_headers)
 
         # Call the backend API
         res_base64 = client.predict(
@@ -222,7 +222,7 @@ def generate_image_handler(x, negative_prompt, fine_edge, fix_perspective, grow_
 
 # --- Gradio UI ---
 
-with gr.Blocks(title="MagicQuill V2", head=zerogpu_fix_js) as demo:
+with gr.Blocks(title="MagicQuill V2") as demo:
     with gr.Row(elem_classes="row"):
         text = gr.Markdown(
         """