Upload 2 files

Dockerfile

@@ -21,7 +21,7 @@ RUN pip install --no-cache-dir -r requirements.txt
 
 # Copy our API entrypoint and the uploaded checkpoint file
 COPY app.py .
-COPY model_imdb_cross_person_4.24_99.46.pth.tar .
+COPY *.pth.tar ./
 
 # Hugging Face Spaces use port 7860
 EXPOSE 7860

app.py

@@ -49,10 +49,12 @@ def get_predictor():
     )
 
     # 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.24_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.")
+    # We auto-detect any .pth.tar file uploaded to the Space.
+    import glob
+    checkpoint_files = glob.glob("*.pth.tar")
+    if not checkpoint_files:
+        raise FileNotFoundError("No checkpoint file ending in .pth.tar found in Space root! Please upload it.")
+    checkpoint_path = checkpoint_files[0]
 
     # Build MiVOLO config
     config = argparse.Namespace(