uartimcs commited on
Commit
394bc91
·
verified ·
1 Parent(s): 3a270b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from donut import DonutModel
9
  def demo_process(pdf_path):
10
  global model, task_prompt, task_name
11
  input_img_list = convert_from_path(pdf_path.name)
12
- input_img = Image.fromarray(input_img_list[0])
13
  output = model.inference(image=input_img, prompt=task_prompt)["predictions"][0]
14
  return output
15
 
 
9
  def demo_process(pdf_path):
10
  global model, task_prompt, task_name
11
  input_img_list = convert_from_path(pdf_path.name)
12
+ input_img = input_img_list[0]
13
  output = model.inference(image=input_img, prompt=task_prompt)["predictions"][0]
14
  return output
15