Update app.py
Browse files
app.py
CHANGED
|
@@ -24,7 +24,8 @@ def extract_chart_data(img: Image.Image, header_text:str):
|
|
| 24 |
|
| 25 |
inputs = processor(images=img, text=header_text, return_tensors="pt").to(device)
|
| 26 |
outputs = model.generate(**inputs)
|
| 27 |
-
|
|
|
|
| 28 |
result = processor.decode(outputs[0], skip_special_tokens=True)
|
| 29 |
return result
|
| 30 |
|
|
|
|
| 24 |
|
| 25 |
inputs = processor(images=img, text=header_text, return_tensors="pt").to(device)
|
| 26 |
outputs = model.generate(**inputs)
|
| 27 |
+
all_results = [processor.decode(seq, skip_special_tokens=True) for seq in outputs]
|
| 28 |
+
print(all_results)
|
| 29 |
result = processor.decode(outputs[0], skip_special_tokens=True)
|
| 30 |
return result
|
| 31 |
|