Update app.py

app.py

@@ -13,24 +13,18 @@ model = VisionEncoderDecoderModel.from_pretrained("./donut-base-finetuned-inv")
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)
 
-def send_to_telegram(message):
-
-    apiToken = os.getenv('TELEGRAM_BOT_TOKEN')
-    chatID = os.getenv('TELEGRAM_CHANNEL_ID')
-    apiURL = f'https://api.telegram.org/bot{apiToken}/sendMessage'
-    try:
-        response = requests.post(apiURL, json={'chat_id': chatID, 'text': message})
-        #print(response.text)
-    #except Exception as e:
-        #print(e)
-
 def process_document(image):
     #save uploaded file locally
     im1 = Image.fromarray(image)
     im1 = im1.save("./geeks.jpg")
 
     #send notification through telegram
-    send_to_telegram("Hello from HuggingFace!")
+    message="Hello from HuggingFace!"
+    apiToken = os.getenv('TELEGRAM_BOT_TOKEN')
+    chatID = os.getenv('TELEGRAM_CHANNEL_ID')
+    apiURL = f'https://api.telegram.org/bot{apiToken}/sendMessage'
+    try:
+        response = requests.post(apiURL, json={'chat_id': chatID, 'text': message})
     
     # prepare encoder inputs
     pixel_values = processor(image, return_tensors="pt").pixel_values