Update app.py

app.py

@@ -12,18 +12,19 @@ model = model.to("cuda")
 
 # Define the function to generate the report
 @spaces.GPU
-def generate_report(image):
+def ask_question(question, image):
     image = Image.open(image).convert("RGB")
-    response = model.write_radiologic_report(image)
+    response = model.ask_question(question, image)
     return response
 
 # Create the Gradio interface
 interface = gr.Interface(
-    fn=generate_report,
-    inputs=gr.Image(type="filepath", label="Upload Image"),
+    fn=ask_question,
+    inputs=[gr.Textbox(lines=1, placeholder="Enter your question here", label="Question"),
+            gr.Image(type="filepath", label="Upload Image")],
     outputs=gr.Textbox(label="Report"),
     title="CXR Report Creator",
-    description="Upload an image to use with the CXR-LLAVA-v2 model."
+    description="Upload an image and enter a question to use with the CXR-LLAVA-v2 model."
 )
 
 # Launch the interface with API enabled