uartimcs commited on
Commit
583d619
·
verified ·
1 Parent(s): 1bfbcb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import gradio as gr
 
2
  import argparse
3
  import torch
4
  from PIL import Image
@@ -7,7 +8,7 @@ from donut import DonutModel
7
 
8
  def demo_process(pdf_path):
9
  global model, task_prompt, task_name
10
- input_img = convert_from_path(pdf_path)
11
  input_img = Image.fromarray(input_img)
12
  output = model.inference(image=input_img, prompt=task_prompt)["predictions"][0]
13
  return output
 
1
  import gradio as gr
2
+ import os
3
  import argparse
4
  import torch
5
  from PIL import Image
 
8
 
9
  def demo_process(pdf_path):
10
  global model, task_prompt, task_name
11
+ input_img = convert_from_path(pdf_path.name)
12
  input_img = Image.fromarray(input_img)
13
  output = model.inference(image=input_img, prompt=task_prompt)["predictions"][0]
14
  return output