uartimcs commited on
Commit
0a0fa4c
·
verified ·
1 Parent(s): 6557e4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ def demo_process(pdf_file):
11
  renderer = PageRenderer()
12
  image = renderer.render_page(page_1)
13
  image_data = image.data
14
- output = model.inference(image=image_data, prompt=task_prompt)["predictions"][0]
15
  return output
16
 
17
  model = DonutModel.from_pretrained("uartimcs/donut-invoice-extract")
 
11
  renderer = PageRenderer()
12
  image = renderer.render_page(page_1)
13
  image_data = image.data
14
+ output = model.inference(image=image, prompt=task_prompt)["predictions"][0]
15
  return output
16
 
17
  model = DonutModel.from_pretrained("uartimcs/donut-invoice-extract")