Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 
 import torch
 from transformers import DonutProcessor, VisionEncoderDecoderModel
+from PIL import Image 
 
 processor = DonutProcessor.from_pretrained("./donut-base-finetuned-inv")
 model = VisionEncoderDecoderModel.from_pretrained("./donut-base-finetuned-inv")
@@ -11,6 +12,10 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)
 
 def process_document(image):
+    #save uploaded file locally
+    im1 = Image.open(image)
+    im1 = im1.save("geeks.jpg")
+
     # prepare encoder inputs
     pixel_values = processor(image, return_tensors="pt").pixel_values