Update app.py

app.py

@@ -15,28 +15,30 @@ def demo_process(input_pdf):
 
     temp_dir = "images"
     basename = os.path.basename(input_pdf).split('.')[0]
-    image_name = basename + "jpg"
+    image_name = basename + ".jpg"
     os.makedirs(temp_dir, exist_ok=True)
     
     with open(os.path.join(temp_dir, image_name), "wb") as f:
         f.write(image_bytes)
+        
+
 
-    zip_path = "images.zip"
+    # zip_path = "images.zip"
     image_path = os.path.join(temp_dir, image_name)
-    
-    with zipfile.ZipFile(zip_path, "w") as zipf:
-        zipf.write(os.path.join(temp_dir, image_name), image_name)
+    output = model.inference(image=image_path, prompt=task_prompt)["predictions"][0]    
+    # with zipfile.ZipFile(zip_path, "w") as zipf:
+    #     zipf.write(os.path.join(temp_dir, image_name), image_name)
         
     os.remove(image_path)
     os.rmdir(temp_dir)
     
-    return zip_path
-
-# task_name = "SGSInvoice"
-# task_prompt = f"<s_{task_name}>"
+    return output
+    
 
-# model = DonutModel.from_pretrained("uartimcs/donut-invoice-extract")
-# model.eval()
+task_name = "SGSInvoice"
+task_prompt = f"<s_{task_name}>"
+model = DonutModel.from_pretrained("uartimcs/donut-invoice-extract")
+model.eval()
 
 iface = gr.Interface(
     fn=demo_process,