Update app.py
Browse files
app.py
CHANGED
|
@@ -24,6 +24,7 @@ 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 |
result = processor.decode(outputs[0], skip_special_tokens=True)
|
| 28 |
return result
|
| 29 |
|
|
@@ -40,7 +41,7 @@ iface = gr.Interface(
|
|
| 40 |
) # Text input
|
| 41 |
],
|
| 42 |
outputs="text",
|
| 43 |
-
title="DePlot Chart Data Extractor
|
| 44 |
description="Upload a chart image and get its extracted data/description."
|
| 45 |
)
|
| 46 |
|
|
|
|
| 24 |
|
| 25 |
inputs = processor(images=img, text=header_text, return_tensors="pt").to(device)
|
| 26 |
outputs = model.generate(**inputs)
|
| 27 |
+
print(outputs)
|
| 28 |
result = processor.decode(outputs[0], skip_special_tokens=True)
|
| 29 |
return result
|
| 30 |
|
|
|
|
| 41 |
) # Text input
|
| 42 |
],
|
| 43 |
outputs="text",
|
| 44 |
+
title="DePlot Chart Data Extractor 2",
|
| 45 |
description="Upload a chart image and get its extracted data/description."
|
| 46 |
)
|
| 47 |
|