Update app.py

app.py

@@ -1,17 +1,13 @@
 import gradio as gr
 import argparse
 import torch
-from PIL import Image
 from donut import DonutModel
 from pdf2image import convert_from_path
-from io import BytesIO
-
-
 
 def demo_process(pdf_file):
-    images = convert_from_path(pdf_file.name)    
+    # images = convert_from_path(pdf_file)    
     # output = model.inference(image=input_img, prompt=task_prompt)["predictions"][0]
-    return str(len(images))
+    return str(len([1,2,3,4,5]))
 
 # model = DonutModel.from_pretrained("uartimcs/donut-invoice-extract")
 # model.eval()