Spaces:

fffiloni
/

MEMO

Running on L40S

fffiloni commited on 20 days ago

Commit

4e552db

•

1 Parent(s): ed73630

Update memo/utils/vision_utils.py

Files changed (1) hide show

memo/utils/vision_utils.py CHANGED Viewed

@@ -65,7 +65,7 @@ def preprocess_image(face_analysis_model: str, image_path: str, image_size: int
     # Initialize the FaceAnalysis model
     face_analysis = FaceAnalysis(
-        name='antelopev2',
         root=face_analysis_model,
         providers=["CUDAExecutionProvider", "CPUExecutionProvider"],
     )

     # Initialize the FaceAnalysis model
     face_analysis = FaceAnalysis(
+       # name='',
         root=face_analysis_model,
         providers=["CUDAExecutionProvider", "CPUExecutionProvider"],
     )