Update app.py

app.py

@@ -2,9 +2,12 @@ import gradio as gr
 import argparse
 import torch
 from PIL import Image
+from pdf2image import convert_from_path
 from donut import DonutModel
-def demo_process(input_img):
+
+def demo_process(pdf_path):
     global model, task_prompt, task_name
+    input_img = convert_from_path(pdf_path)
     input_img = Image.fromarray(input_img)
     output = model.inference(image=input_img, prompt=task_prompt)["predictions"][0]
     return output
@@ -17,5 +20,5 @@ task_prompt = f"<s_{task_name}>"
 
 model = DonutModel.from_pretrained("uartimcs/donut-invoice-extract")
 model.eval()
-demo = gr.Interface(fn=demo_process,inputs="image",outputs="json", title=f"Donut 🍩 demonstration for `{task_name}` task",)
+demo = gr.Interface(fn=demo_process,inputs=gr.inputs.File(label="Upload PDF"),outputs="json", title=f"Donut 🍩 demonstration for `{task_name}` task",)
 demo.launch()