Update app.py
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ model.to(device)
|
|
| 20 |
def extract_chart_data(img: Image.Image, header_text:str):
|
| 21 |
# Provide a header prompt for VQA
|
| 22 |
# header_text = "Convert this bar chart into a table of quarter and cost values"
|
| 23 |
-
|
| 24 |
|
| 25 |
inputs = processor(images=img, text=header_text, return_tensors="pt").to(device)
|
| 26 |
outputs = model.generate(**inputs)
|
|
|
|
| 20 |
def extract_chart_data(img: Image.Image, header_text:str):
|
| 21 |
# Provide a header prompt for VQA
|
| 22 |
# header_text = "Convert this bar chart into a table of quarter and cost values"
|
| 23 |
+
print("Header text:", header_text)
|
| 24 |
|
| 25 |
inputs = processor(images=img, text=header_text, return_tensors="pt").to(device)
|
| 26 |
outputs = model.generate(**inputs)
|