Update app.py

app.py

@@ -66,16 +66,33 @@ def inference(image):
 
     return None  # In case of unexpected errors
 
-# Create Gradio interface
-demo = gr.Interface(
-    fn=inference,
-    inputs=[
-        gr.Image(type="pil", label="Upload Image")
-    ],
-    outputs=gr.Textbox(label="Response"),
-    title="Image Analysis AI",
-    description="Upload an image and ask a question about it. The AI will analyze and respond.",
-)
+# Create Gradio interface using Blocks
+with gr.Blocks() as demo:
+    gr.Markdown("# BugsBunny-LLama-3.2-11B-Base-Medium Demo")
+
+    with gr.Row():
+        # Container for the image takes full width
+        with gr.Column(scale=1):
+            image_input = gr.Image(
+                type="pil",
+                label="Upload Image",
+                elem_id="large-image",
+                height=500,  # Increased height for larger display
+            )
+
+    with gr.Row():
+        # Container for the text output takes full width
+        with gr.Column(scale=1):
+            text_output = gr.Textbox(
+                label="Response",
+                elem_id="response-text",
+                lines=25,
+                max_lines=10,
+            )
+
+    # Button to trigger the analysis
+    submit_btn = gr.Button("Analyze Image", variant="primary")
+    submit_btn.click(fn=inference, inputs=[image_input], outputs=[text_output])
+
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()