Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import os
+import shutil
 import argparse
 import torch
 from PIL import Image
@@ -9,8 +10,12 @@ from donut import DonutModel
 
 def demo_process(doc):
     global model, task_prompt, task_name
-    output = os.getcwd() + " file " + doc
-    # output = model.inference(image=img, prompt=task_prompt)["predictions"][0]
+    file_name = os.path.basename(doc)
+    new_file_location = os.path.join(os.getcwd(),file_name)
+    
+    shutil.copyfile(doc, new_file_location)
+    img = convert_from_path(new_file_location)[0]
+    output = model.inference(image=img, prompt=task_prompt)["predictions"][0]
     return output
     
 parser = argparse.ArgumentParser()
@@ -22,5 +27,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()