uartimcs commited on
Commit
11927ca
·
verified ·
1 Parent(s): 8a655f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,7 +25,7 @@ args, left_argv = parser.parse_known_args()
25
  task_name = args.task
26
  task_prompt = f"<s_{task_name}>"
27
 
28
- # model = DonutModel.from_pretrained("uartimcs/donut-invoice-extract")
29
- # model.eval()
30
  demo = gr.Interface(fn=demo_process,inputs=gr.File(label="Upload PDF"),outputs="json", title=f"Donut 🍩 demonstration for `{task_name}` task",)
31
  demo.launch()
 
25
  task_name = args.task
26
  task_prompt = f"<s_{task_name}>"
27
 
28
+ model = DonutModel.from_pretrained("uartimcs/donut-invoice-extract")
29
+ model.eval()
30
  demo = gr.Interface(fn=demo_process,inputs=gr.File(label="Upload PDF"),outputs="json", title=f"Donut 🍩 demonstration for `{task_name}` task",)
31
  demo.launch()