Update app_v4.py

app_v4.py

@@ -273,11 +273,11 @@ with gr.Blocks(title="FLUX Turbo Upscaler", fill_height=True) as demo:
     control_image.upload(
         generate_caption,
         inputs=[control_image],
-        outputs=[caption_state]
+        outputs=[caption_state, original_dimensions]
     ).then(
         generate_focus,
         inputs=[control_image, focus],
-        outputs=[focus_state]
+        outputs=[focus_state, original_dimensions]
     ).then(
         combine_caption_focus,
         inputs=[caption_state, focus_state],
@@ -286,11 +286,11 @@ with gr.Blocks(title="FLUX Turbo Upscaler", fill_height=True) as demo:
     caption_button.click(
         fn=generate_caption,
         inputs=[control_image],
-        outputs=[prompt]
+        outputs=[prompt, original_dimensions]
     ).then(
         generate_focus,
         inputs=[control_image, focus],
-        outputs=[focus_state]
+        outputs=[focus_state, original_dimensions]
     ).then(
         combine_caption_focus,
         inputs=[caption_state, focus_state],