Update app.py

app.py

@@ -10,8 +10,9 @@ def demo_process(pdf_path):
     global model, task_prompt, task_name
     input_img_list = convert_from_path(pdf_path.name)
     input_img = input_img_list[0]
-    output = model.inference(image=input_img, prompt=task_prompt)["predictions"][0]
-    return output
+    return type(input_img)
+    # output = model.inference(image=input_img, prompt=task_prompt)["predictions"][0]
+    # return output
     
 parser = argparse.ArgumentParser()
 parser.add_argument("--task", type=str, default="SGSInvoice")
@@ -22,5 +23,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="json", title=f"Donut 🍩 demonstration for `{task_name}` task",)
+demo = gr.Interface(fn=demo_process,inputs=gr.File(label="Upload PDF"),outputs="text", title=f"Donut 🍩 demonstration for `{task_name}` task",)
 demo.launch()