Upload app.py

app.py

@@ -42,22 +42,22 @@ def get_predictor():
     from huggingface_hub import hf_hub_download
     from mivolo.predictor import Predictor
 
-    # Download the YOLOv8 person+face detector weights
+    # Download the YOLOv8 person+face detector weights from the public working repo
     detector_weights = hf_hub_download(
-        repo_id="WildChlamydia/MiVOLO_D1",
+        repo_id="iitolstykh/demo_yolov8_detector",
         filename="yolov8x_person_face.pt"
     )
 
-    # Download the MiVOLO age+gender estimator weights
-    checkpoint = hf_hub_download(
-        repo_id="WildChlamydia/MiVOLO_D1",
-        filename="mivolo_d1_age_gender.pth.tar"
-    )
+    # The age/gender checkpoint is no longer publicly hosted on HF.
+    # We expect the user to have uploaded it directly to the HF Space.
+    checkpoint_path = "model_imdb_cross_person_4.22_99.46.pth.tar"
+    if not os.path.exists(checkpoint_path):
+        raise FileNotFoundError(f"Checkpoint {checkpoint_path} not found in Space root! Please upload it.")
 
     # Build MiVOLO config
     config = argparse.Namespace(
         detector_weights=detector_weights,
-        checkpoint=checkpoint,
+        checkpoint=checkpoint_path,
         device="cpu",
         with_persons=True,     # Use full-body context for better accuracy
         disable_faces=False,   # Also use face features