Update app.py

app.py

@@ -17,9 +17,9 @@ model.to(device)
 #     result = processor.decode(outputs[0], skip_special_tokens=True)
 #     return result
 
-def extract_chart_data(img: Image.Image):
+def extract_chart_data(img: Image.Image, header_text:str):
     # Provide a header prompt for VQA
-    header_text = "Convert this bar chart into a table of quarter and cost values"
+    # header_text = "Convert this bar chart into a table of quarter and cost values"
     
     inputs = processor(images=img, text=header_text, return_tensors="pt").to(device)
     outputs = model.generate(**inputs)
@@ -30,9 +30,16 @@ def extract_chart_data(img: Image.Image):
 # Gradio interface
 iface = gr.Interface(
     fn=extract_chart_data,
-    inputs=gr.Image(type="pil"),
+    # inputs=gr.Image(type="pil"),
+    inputs=[
+        gr.Image(type="pil", label="Upload Chart Image"),  # Image input
+        gr.Textbox(
+            label="Instruction / Header Text",
+            placeholder="Enter instructions like 'Convert this bar chart into a table of quarter and cost values'"
+        )  # Text input
+    ],
     outputs="text",
-    title="DePlot Chart Data Extractor 1",
+    title="DePlot Chart Data Extractor 2",
     description="Upload a chart image and get its extracted data/description."
 )