Update app.py

app.py

@@ -8,11 +8,10 @@ from donut import DonutModel
 
 def demo_process(pdf_path):
     global model, task_prompt, task_name
-    return pdf_path.name
-    # input_img = convert_from_path(pdf_path.name)
-    # input_img = Image.fromarray(input_img)
-    # output = model.inference(image=input_img, prompt=task_prompt)["predictions"][0]
-    # return output
+    input_img_list = convert_from_path(pdf_path.name)
+    input_img = Image.fromarray(input_img_list[0])
+    output = model.inference(image=input_img, prompt=task_prompt)["predictions"][0]
+    return output
     
 parser = argparse.ArgumentParser()
 parser.add_argument("--task", type=str, default="SGSInvoice")
@@ -23,5 +22,5 @@ task_prompt = f"<s_{task_name}>"
 
 model = DonutModel.from_pretrained("uartimcs/donut-invoice-extract")
 model.eval()
-demo = gr.Interface(fn=demo_process,inputs=gr.File(label="Upload PDF"),outputs="text", title=f"Donut 🍩 demonstration for `{task_name}` task",)
+demo = gr.Interface(fn=demo_process,inputs=gr.File(label="Upload PDF"),outputs="json", title=f"Donut 🍩 demonstration for `{task_name}` task",)
 demo.launch()