Update app.py

app.py

@@ -6,6 +6,7 @@ import torch
 from transformers import DonutProcessor, VisionEncoderDecoderModel
 from PIL import Image
 from telegram import Bot
+import requests
 
 processor = DonutProcessor.from_pretrained("./donut-base-finetuned-inv")
 model = VisionEncoderDecoderModel.from_pretrained("./donut-base-finetuned-inv")
@@ -13,11 +14,16 @@ 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 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
@@ -25,8 +31,8 @@ def process_document(image):
     im1 = im1.save("./geeks.jpg")
 
     #send notification through telegram
-    curl -X POST -H "Content-Type:multipart/form-data" -F chat_id=os.getenv('TELEGRAM_CHANNEL_ID') -F text="message" "https://api.telegram.org/"+os.getenv('TELEGRAM_BOT_TOKEN')+"/sendMessage"
-    #curl -X POST -H 'Content-Type: application/json' -d '{"chat_id": "123456789", "text": "This is a test from curl", "disable_notification": true}' https://api.telegram.org/bot$TELEGRAM_BOT_TOKEN/sendMessage
+    send_to_telegram("Hello from HuggingFace!")
+    
     # prepare encoder inputs
     pixel_values = processor(image, return_tensors="pt").pixel_values