Update app.py

app.py

@@ -19,22 +19,22 @@ model.to(device)
 def filter_output(output):
     return output.replace("<0x0A>", "")
 
-def summarize_chart(image):
-    inputs = processor(images=image, text="", return_tensors="pt").to(device)
+def chart_qa(image, question):
+    inputs = processor(images=image, text=question, return_tensors="pt").to(device)
     predictions = model.generate(**inputs, max_new_tokens=512)
     return filter_output(processor.decode(predictions[0], skip_special_tokens=True))
 
    
 image = gr.inputs.Image(type="pil", label="Chart")
-answer = gr.outputs.Textbox(label="Chart Summary")
+answer = gr.outputs.Textbox(label="Model Output")
 examples = [["chart_example.png", "Which country has the second highest death rate?"], ]
             #["chart_example_2.png"], ["chart_example_3.png"], ["chart_example_4.png"]]
 
 title = "Interactive demo: chart QA"
 description = "Gradio Demo for matcha model, fine-tuned on the ChartQA dataset. To use it, simply upload your image and click 'submit', or click one of the examples to load them."
 
-interface = gr.Interface(fn=summarize_chart, 
-                         inputs=[image], 
+interface = gr.Interface(fn=chart_qa, 
+                         inputs=[image, question], 
                          outputs=answer, 
                          examples=examples, 
                          title=title,