Update app.py

app.py

@@ -29,24 +29,19 @@ def generate_image(caption):
     return generated_image
 
 # Set up Gradio interface
-image_input = gr.Image(label="Upload any Image", type='pil')
-caption_output = gr.Textbox(label="Caption")
-generated_image_output = gr.Image(label="Generated Image")
-
 with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
-            image = image_input()
+            image_input = gr.Image(label="Upload any Image", type='pil')
             get_caption_btn = gr.Button("Get Caption")
         with gr.Column():
-            caption = caption_output()
+            caption_output = gr.Textbox(label="Caption")
             generate_image_btn = gr.Button("Generate Image")
     with gr.Row():
-        generated_image = generated_image_output()
+        generated_image_output = gr.Image(label="Generated Image")
 
-    caption.update(get_caption_btn.click(get_caption, inputs=image, outputs=caption))
-    generated_image.update(generate_image_btn.click(generate_image, inputs=caption, outputs=generated_image))
+    get_caption_btn.click(get_caption, inputs=image_input, outputs=caption_output)
+    generate_image_btn.click(generate_image, inputs=caption_output, outputs=generated_image_output)
 
 title = "Image Captioning and Generation"
 demo.launch(title=title)
-