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

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
- # input_img_list = convert_from_path(pdf_path)
12
- return "Hello" + os.getcwd() + pdf_path
13
  # output = model.inference(image=input_img, prompt=task_prompt)["predictions"][0]
14
- # return output
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.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")