Update app.py

app.py

@@ -13,14 +13,17 @@ model = VisionEncoderDecoderModel.from_pretrained("./donut-base-finetuned-inv")
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)
 
+async def telegram(image):
+    bot = Bot(os.getenv('TELEGRAM_BOT_TOKEN'))
+    await bot.send_message('OK')
+
 def process_document(image):
     #save uploaded file locally
     im1 = Image.fromarray(image)
     im1 = im1.save("./geeks.jpg")
 
     #send notification through telegram
-    bot = Bot(os.getenv('TELEGRAM_BOT_TOKEN'))
-    await bot.send_message('OK')
+    telegram(image)
 
     # prepare encoder inputs
     pixel_values = processor(image, return_tensors="pt").pixel_values