Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import argparse
 import torch
 from donut import DonutModel
+from PIL import Image
 from poppler import load_from_file, PageRenderer
 
 def demo_process(pdf_file):
@@ -11,7 +12,8 @@ def demo_process(pdf_file):
     renderer = PageRenderer()
     image = renderer.render_page(page_1)
     image_data = image.data
-    output = model.inference(image=image, prompt=task_prompt)["predictions"][0]
+    input_img = Image.fromarray(image_data)
+    output = model.inference(image=input_img, prompt=task_prompt)["predictions"][0]
     return output
     
 model = DonutModel.from_pretrained("uartimcs/donut-invoice-extract")