Update app.py

app.py

@@ -5,10 +5,9 @@ from donut import DonutModel
 from pdf2image import convert_from_path
 
 def demo_process(pdf_file):
-    # images = convert_from_path(pdf_file)    
+    images = convert_from_path(pdf_file.name) 
     # output = model.inference(image=input_img, prompt=task_prompt)["predictions"][0]
-    return pdf_file.name
-
+    return str(len(images))
 # model = DonutModel.from_pretrained("uartimcs/donut-invoice-extract")
 # model.eval()
 task_name = "SGSInvoice"