uartimcs commited on
Commit
f70b02d
·
verified ·
1 Parent(s): 7079d89

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,10 +8,10 @@ from donut import DonutModel
8
 
9
  def demo_process(pdf_path):
10
  global model, task_prompt, task_name
11
- full_path = os.getcwd() + pdf_path
12
- input_img_list = convert_from_path(full_path)
13
  # output = model.inference(image=input_img, prompt=task_prompt)["predictions"][0]
14
- return "Hello World"
15
 
16
  parser = argparse.ArgumentParser()
17
  parser.add_argument("--task", type=str, default="SGSInvoice")
 
8
 
9
  def demo_process(pdf_path):
10
  global model, task_prompt, task_name
11
+ full_path = os.path.join(os.getcwd(),pdf_path)
12
+ # input_img_list = convert_from_path(full_path)
13
  # output = model.inference(image=input_img, prompt=task_prompt)["predictions"][0]
14
+ return full_path
15
 
16
  parser = argparse.ArgumentParser()
17
  parser.add_argument("--task", type=str, default="SGSInvoice")