Update app.py

app.py

@@ -84,4 +84,22 @@ try:
         ],
         outputs=[
             gr.Textbox(label="Response", placeholder="Model response will appear here..."),
-            gr.Image(label="Gene
+            gr.Image(label="Generated Image", type="pil")
+        ],
+        title="LLAVA Model - Medical Image and Question",
+        description="Upload a medical image and ask a specific question about the image for a medical description.",
+        additional_inputs=[
+            gr.Textbox(label="System message", value="You are a friendly Chatbot."),
+            gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
+            gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
+            gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)")
+        ]
+    )
+
+    # Launch the Gradio interface
+    if __name__ == "__main__":
+        print("Launching Gradio interface...")
+        demo.launch()
+
+except Exception as e:
+    print(f"Error during Gradio setup: {str(e)}")