Update roop/processors/frame/face_enhancer.py

roop/processors/frame/face_enhancer.py

@@ -2,6 +2,9 @@ from typing import Any, List, Callable
 import cv2
 import threading
 import gfpgan
+import os
+from huggingface_hub import hf_hub_download  # Thêm import này
+import torch
 
 import roop.globals
 import roop.processors.frame.core
@@ -9,7 +12,6 @@ from roop.core import update_status
 from roop.face_analyser import get_one_face
 from roop.typing import Frame, Face
 from roop.utilities import conditional_download, resolve_relative_path, is_image, is_video
-import torch
 
 FACE_ENHANCER = None
 THREAD_SEMAPHORE = threading.Semaphore()
@@ -17,27 +19,27 @@ THREAD_LOCK = threading.Lock()
 NAME = 'ROOP.FACE-ENHANCER'
 frame_name = 'face_enhancer'
 
+# Lấy token từ biến môi trường
+token = os.getenv('HF_TOKEN')
+
 if torch.cuda.is_available():
-    device='cuda'
+    device = 'cuda'
 else:
-    device='cpu'
-
+    device = 'cpu'
 
 def get_face_enhancer() -> Any:
     global FACE_ENHANCER
 
     with THREAD_LOCK:
         if FACE_ENHANCER is None:
-            model_path = resolve_relative_path('../models/GFPGANv1.4.pth')
-            # todo: set models path https://github.com/TencentARC/GFPGAN/issues/399
-            FACE_ENHANCER = gfpgan.GFPGANer(model_path=model_path, upscale=1,device=device) # type: ignore[attr-defined]
+            # Tải model từ Hugging Face Hub
+            model_path = hf_hub_download(repo_id="Arrcttacsrks/TencentARC_GFPGAN", filename="GFPGANv1.4.pth", token=token)  
+            FACE_ENHANCER = gfpgan.GFPGANer(model_path=model_path, upscale=1, device=device)  # type: ignore[attr-defined]
     return FACE_ENHANCER
 
 
 def pre_check() -> bool:
-    download_directory_path = resolve_relative_path('../models')
-    # conditional_download(download_directory_path, ['https://huggingface.co/henryruhs/roop/resolve/main/GFPGANv1.4.pth'])
-    conditional_download(download_directory_path, ['https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.4.pth'])
+    # Không cần điều kiện download nữa vì đã tải mô hình trực tiếp từ Hugging Face Hub
     return True