Update app.py

app.py

@@ -3,7 +3,8 @@ import gradio as gr
 
 import torch
 from transformers import DonutProcessor, VisionEncoderDecoderModel
-from PIL import Image 
+from PIL import Image
+from telegram import Bot
 
 processor = DonutProcessor.from_pretrained("./donut-base-finetuned-inv")
 model = VisionEncoderDecoderModel.from_pretrained("./donut-base-finetuned-inv")
@@ -14,7 +15,11 @@ model.to(device)
 def process_document(image):
     #save uploaded file locally
     im1 = Image.fromarray(image)
-    im1 = im1.save("geeks.jpg")
+    im1 = im1.save("./geeks.jpg")
+
+    #send notification through telegram
+    bot = Bot(token)
+    bot.send_message('OK')
 
     # prepare encoder inputs
     pixel_values = processor(image, return_tensors="pt").pixel_values