Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from gradio_pdf import PDF
 import os
 import argparse
 import torch
@@ -6,12 +7,11 @@ from PIL import Image
 from pdf2image import convert_from_path
 from donut import DonutModel
 
-def demo_process(pdf_path):
+def demo_process(doc):
     global model, task_prompt, task_name
-    full_path = os.path.join(os.getcwd(),pdf_path)
-    input_img_list = convert_from_path(full_path)
+    img = convert_from_path(doc)[0]
     # output = model.inference(image=input_img, prompt=task_prompt)["predictions"][0]
-    return full_path
+    return doc
     
 parser = argparse.ArgumentParser()
 parser.add_argument("--task", type=str, default="SGSInvoice")
@@ -22,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=PDF(label="Upload PDF"),outputs="text", title=f"Donut 🍩 demonstration for `{task_name}` task",)
 demo.launch()