Update app.py

app.py

@@ -44,7 +44,6 @@ def respond(
                 image_b64 = image_to_base64(img)
                 messages.append({"role": "user", "content": "Image uploaded", "image": image_b64})
 
-    # Call Hugging Face model for response
     try:
         responses = []
         generated_image = None
@@ -67,10 +66,10 @@ def respond(
                 # Optionally convert to RGB if needed
                 # generated_image = generated_image.convert("RGB")
 
-        yield responses, generated_image
+        return responses, generated_image
 
     except Exception as e:
-        yield [str(e)], None
+        return [str(e)], None
 
 # Debugging print statements
 print("Starting Gradio interface setup...")
@@ -84,10 +83,10 @@ try:
         ],
         outputs=[
             gr.Textbox(label="Response", placeholder="Model response will appear here..."),
-            gr.Image(label="Generated Image", type="pil")
+            gr.Image(label="Generated Image", type="pil", output=True)
         ],
-        title="LLAVA Model - Medical Image and Question",
-        description="Upload a medical image and ask a specific question about the image for a medical description.",
+        title="LLAVA Model - Medical Image and Text",
+        description="Upload a medical image and ask a specific question about the image or provide text input 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"),