update url

facelib/detection/__init__.py

@@ -25,10 +25,10 @@ def init_detection_model(model_name, half=False, device='cuda'):
 def init_retinaface_model(model_name, half=False, device='cuda'):
     if model_name == 'retinaface_resnet50':
         model = RetinaFace(network_name='resnet50', half=half)
-        model_url = 'https://github.com/jnjaby/KEEP/releases/download/v0.1.0/detection_Resnet50_Final.pth'
+        model_url = 'https://github.com/jnjaby/KEEP/releases/download/v1.0.0/detection_Resnet50_Final.pth'
     elif model_name == 'retinaface_mobile0.25':
         model = RetinaFace(network_name='mobile0.25', half=half)
-        model_url = 'https://github.com/jnjaby/KEEP/releases/download/v0.1.0/detection_mobilenet0.25_Final.pth'
+        model_url = 'https://github.com/jnjaby/KEEP/releases/download/v1.0.0/detection_mobilenet0.25_Final.pth'
     else:
         raise NotImplementedError(f'{model_name} is not implemented.')
 
@@ -49,10 +49,10 @@ def init_retinaface_model(model_name, half=False, device='cuda'):
 def init_yolov5face_model(model_name, device='cuda'):
     if model_name == 'YOLOv5l':
         model = YoloDetector(config_name='facelib/detection/yolov5face/models/yolov5l.yaml', device=device)
-        model_url = 'https://github.com/jnjaby/KEEP/releases/download/v0.1.0/yolov5l-face.pth'
+        model_url = 'https://github.com/jnjaby/KEEP/releases/download/v1.0.0/yolov5l-face.pth'
     elif model_name == 'YOLOv5n':
         model = YoloDetector(config_name='facelib/detection/yolov5face/models/yolov5n.yaml', device=device)
-        model_url = 'https://github.com/jnjaby/KEEP/releases/download/v0.1.0/yolov5n-face.pth'
+        model_url = 'https://github.com/jnjaby/KEEP/releases/download/v1.0.0/yolov5n-face.pth'
     else:
         raise NotImplementedError(f'{model_name} is not implemented.')
     

facelib/parsing/__init__.py

@@ -8,10 +8,10 @@ from .parsenet import ParseNet
 def init_parsing_model(model_name='bisenet', half=False, device='cuda'):
     if model_name == 'bisenet':
         model = BiSeNet(num_class=19)
-        model_url = 'https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/parsing_bisenet.pth'
+        model_url = 'https://github.com/jnjaby/KEEP/releases/download/v1.0.0/parsing_bisenet.pth'
     elif model_name == 'parsenet':
         model = ParseNet(in_size=512, out_size=512, parsing_ch=19)
-        model_url = 'https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/parsing_parsenet.pth'
+        model_url = 'https://github.com/jnjaby/KEEP/releases/download/v1.0.0/parsing_parsenet.pth'
     else:
         raise NotImplementedError(f'{model_name} is not implemented.')
 

hugging_face/app.py

@@ -239,11 +239,28 @@ download_url_to_file("https://github.com/jnjaby/KEEP/releases/download/media/rea
 download_url_to_file("https://github.com/jnjaby/KEEP/releases/download/media/real_3.mp4", os.path.join(sample_videos_dir, "real_3.mp4"))
 download_url_to_file("https://github.com/jnjaby/KEEP/releases/download/media/real_4.mp4", os.path.join(sample_videos_dir, "real_4.mp4"))
 
-model_dir = os.path.join("/home/user/app/", "weights/KEEP")
-_ = load_file_from_url(url='https://github.com/jnjaby/KEEP/releases/download/v1.0.0/KEEP-b76feb75.pth', model_dir=model_dir, progress=True, file_name=None)
 
-realesrgan_dir = os.path.join("/home/user/app/", "weights/realesrgan")
-_ = load_file_from_url(url='https://github.com/jnjaby/KEEP/releases/download/v1.0.0/RealESRGAN_x2plus.pth', model_dir=realesrgan_dir, progress=True, file_name=None)
+model_dir = "/home/user/app/weights/"
+model_url = "https://github.com/jnjaby/KEEP/releases/download/v1.0.0/"
+
+_ = load_file_from_url(url=os.path.join(model_url, 'KEEP-b76feb75.pth'),
+                        model_dir=os.path.join(model_dir, "KEEP"), progress=True, file_name=None)
+
+_ = load_file_from_url(url=os.path.join(model_url, 'detection_Resnet50_Final.pth'),
+                        model_dir=os.path.join(model_dir, "facelib"), progress=True, file_name=None)
+_ = load_file_from_url(url=os.path.join(model_url, 'detection_mobilenet0.25_Final.pth'),
+                        model_dir=os.path.join(model_dir, "facelib"), progress=True, file_name=None)
+_ = load_file_from_url(url=os.path.join(model_url, 'yolov5n-face.pth'),
+                        model_dir=os.path.join(model_dir, "facelib"), progress=True, file_name=None)
+_ = load_file_from_url(url=os.path.join(model_url, 'yolov5l-face.pth'),
+                        model_dir=os.path.join(model_dir, "facelib"), progress=True, file_name=None)
+_ = load_file_from_url(url=os.path.join(model_url, 'parsing_parsenet.pth'),
+                        model_dir=os.path.join(model_dir, "facelib"), progress=True, file_name=None)
+
+_ = load_file_from_url(url=os.path.join(model_url, 'RealESRGAN_x2plus.pth'),
+                        model_dir=os.path.join(model_dir, "realesrgan"), progress=True, file_name=None)
+
+
 
 # Launching the Gradio interface.
 demo = gr.Interface(