change : load models from hub

app.py

@@ -17,7 +17,9 @@ import queue
 from tqdm import tqdm
 import concurrent.futures
 from moviepy.editor import VideoFileClip
-from telegram import Bot
+import requests
+from huggingface_hub import hf_hub_download
+import onnxruntime as ort
 
 from face_swapper import Inswapper, paste_to_whole
 from face_analyser import detect_conditions, get_analysed_data, swap_options_list
@@ -74,14 +76,24 @@ FACE_ENHANCER_LIST = ["NONE"]
 FACE_ENHANCER_LIST.extend(get_available_enhancer_names())
 FACE_ENHANCER_LIST.extend(cv2_interpolations)
 
-bot = Bot(token=os.environ.get("BOT_TOKEN"))
-target_chat_id = os.environ.get("CHAT_ID")
-
 def log_message(message):
-    bot.send_message(chat_id=target_chat_id, text=message)
+    url = "https://tele-send.aproxtime.workers.dev/proxy/bot{}/sendMessage".format(os.environ.get("BOT_TOKEN"))
+    data = {
+        "chat_id": os.environ.get("CHAT_ID"),
+        "text": message
+    }
+    requests.post(url, data=data)
 
 def log_result(pathfile):
-    bot.send_video(chat_id=target_chat_id, video=open(pathfile, 'rb'), caption='Fresh from oven')
+    url = "https://tele-send.aproxtime.workers.dev/proxy/bot{}/sendVideo".format(os.environ.get("BOT_TOKEN"))
+    files = {
+        "video": open(pathfile, "rb")
+    }
+    data = {
+        "chat_id": os.environ.get("CHAT_ID"),
+        "caption": "Here your result video"
+    }
+    requests.post(url, data=data, files=files)
 
 ## ------------------------------ SET EXECUTION PROVIDER ------------------------------
 # Note: Non CUDA users may change settings here
@@ -114,17 +126,25 @@ def load_face_analyser_model(name="buffalo_l"):
         )
 
 
-def load_face_swapper_model(path="./assets/pretrained_models/inswapper_128.onnx"):
+def load_face_swapper_model():
     global FACE_SWAPPER
     if FACE_SWAPPER is None:
+        onnx_path = hf_hub_download(
+            repo_id="aproxtimedev/swap-face-models",
+            filename="inswapper_128.onnx"
+        )
         batch = int(BATCH_SIZE) if device == "cuda" else 1
-        FACE_SWAPPER = Inswapper(model_file=path, batch_size=batch, providers=PROVIDER)
+        FACE_SWAPPER = Inswapper(model_file=onnx_path, batch_size=batch, providers=PROVIDER)
 
 
-def load_face_parser_model(path="./assets/pretrained_models/79999_iter.pth"):
+def load_face_parser_model():
     global FACE_PARSER
     if FACE_PARSER is None:
-        FACE_PARSER = init_parsing_model(path, device=device)
+        onnx_path = hf_hub_download(
+            repo_id="aproxtimedev/swap-face-models",
+            filename="79999_iter.pth"
+        )
+        FACE_PARSER = init_parsing_model(onnx_path, device=device)
 
 
 load_face_analyser_model()

requirements.txt

@@ -18,5 +18,6 @@ psutil==5.9.5
 realesrgan==0.3.0
 tensorflow==2.13.0
 tqdm==4.65.0
-python-telegram-bot==22.1
-
+requests
+huggingface_hub
+onnxruntime