Update app.py

app.py

@@ -23,20 +23,18 @@ def process(input_pdf):
     
     image_path = os.path.join(temp_dir, image_name)
     
-    output = model.inference(image=image_path, prompt=task_prompt)["predictions"][0]
-
     os.remove(image_path)
     os.rmdir(temp_dir)
     
-    return output
+    return image_path
 
-task_name = "SGSInvoice"
-task_prompt = f"<s_{task_name}>"
+# task_name = "SGSInvoice"
+# task_prompt = f"<s_{task_name}>"
 
-model = DonutModel.from_pretrained("uartimcs/donut-invoice-extract")
-model.eval()
+# model = DonutModel.from_pretrained("uartimcs/donut-invoice-extract")
+# model.eval()
 
-demo = gr.Interface(fn=process,inputs=gr.File(label="Upload PDF File"),outputs="json", title=f"Donut 🍩 demonstration for `{task_name}` task",)
+demo = gr.Interface(fn=process,inputs=gr.File(label="Upload PDF File"),outputs=gr.File(label="Output File"), title=f"Donut 🍩 demonstration for `{task_name}` task",)
 demo.launch()