Update app.py

app.py

@@ -3,6 +3,8 @@ from fastapi.middleware.cors import CORSMiddleware
 from inference import ObjectDetector
 import numpy as np
 import cv2
+import socket
+import uvicorn
 
 # Configuration
 MODEL_ONNX_PATH = "model.onnx"
@@ -55,6 +57,11 @@ app.add_middleware(
     expose_headers=["*"]
 )
 
+def get_base_url():
+    hostname = socket.gethostname()
+    port = 7860  # Hugging Face Spaces uses port 7860
+    return f"https://{hostname}.hf.space"
+
 @app.options("/detect")
 async def detect_options():
     return {"Allow": "POST"}
@@ -93,5 +100,7 @@ async def detect_objects(file: UploadFile = File(...)):
         raise HTTPException(500, f"Processing error: {str(e)}")
 
 if __name__ == "__main__":
-    import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)  # Hugging Face requires port 7860
+    base_url = get_base_url()
+    print(f"Base URL: {base_url}")
+    print(f"API endpoint: {base_url}/detect")
+    uvicorn.run(app, host="0.0.0.0", port=7860)