aa

Dockerfile

@@ -17,10 +17,10 @@ RUN mkdir -p models
 
 # Expose the port FastAPI will run on. Hugging Face Spaces often expects 
 # port 8000 for web apps.
-EXPOSE 8000
+EXPOSE 7860
 
 # Command to run the Uvicorn server
 # The --host 0.0.0.0 makes the server accessible from outside the 
 # container
 # The --port 7860 matches the EXPOSE instruction
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -3,6 +3,9 @@ from pydantic import BaseModel
 import os
 import joblib
 import pandas as pd
+import threading
+import time
+import requests
 
 # --- 0. Initialize FastAPI app ---
 app = FastAPI()
@@ -87,4 +90,14 @@ async def predict_inference(data: InferenceInput):
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Inference failed: {str(e)}")
 
+def keep_alive():
+    while True:
+        try:
+            requests.get("http://localhost:7860/health")
+        except Exception:
+            pass
+        time.sleep(60)  # Ping every 60 seconds
+
+threading.Thread(target=keep_alive, daemon=True).start()
+
 # --- 5. Run the app ---