Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import fitz  # PyMuPDF
 from PIL import Image
 
-def pdf_to_jpg(pdf_file):
+def analysis_process(pdf_file):
     # Open the PDF file
     document = fitz.open(pdf_file.name)
     # Get the first page
@@ -13,12 +13,20 @@ def pdf_to_jpg(pdf_file):
     image = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
     image_path = "first_page.jpg"
     image.save(image_path)
-    return image_path
+
+    output = model.inference(image=image_path, prompt=task_prompt)["predictions"][0] 
+    return output
+
+task_name = "SGSInvoice"
+task_prompt = f"<s_{task_name}>"
+model = DonutModel.from_pretrained("uartimcs/donut-invoice-extract")
+model.eval()
+
 
 interface = gr.Interface(
-    fn=pdf_to_jpg,
+    fn=analysis_process,
     inputs=gr.File(label="Upload PDF"),
-    outputs=gr.Image(label="First Page JPG")
+    outputs="json"
 )
 
 interface.launch()