ravi-vc commited on
Commit
c577e75
·
verified ·
1 Parent(s): c4bbe3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- print(outputs)
 
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