Add HEALTHCHECK and app_port

.huggingface/space.yaml

@@ -1,3 +1,2 @@
 sdk: docker
-docker:
-  port: 8000
+app_port: 8000

Dockerfile

@@ -23,5 +23,12 @@ RUN mkdir -p /models/clip && \
 RUN python3 -c "from transformers import AutoTokenizer; AutoTokenizer.from_pretrained('bert-base-uncased').save_pretrained('/models/bert-tokenizer')"
 RUN python3 -c "from transformers import CLIPProcessor; CLIPProcessor.from_pretrained('openai/clip-vit-base-patch32').save_pretrained('/models/clip')"
 EXPOSE 8000
+
+# Install curl if it's not already installed
+RUN apt-get update && apt-get install -y curl
+
+# Add the health check
+HEALTHCHECK CMD curl --fail http://localhost:8000/ || exit 1
+
 #CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000", "--log-level", "debug"]

main.py

@@ -20,9 +20,9 @@ app.mount("/static", StaticFiles(directory=static_dir), name="static")
 # async def landing():
 #     return Path("vit_captioning/static/landing.html").read_text()
 
-@app.get("/") # DEBUGGING
-def health_check():
-    return {"status": "ok"}
+@app.get("/")
+def read_root():
+    return {"message": "✅ Space is working"}
 
 # ✅ Captioning page at `/captioning`
 @app.get("/captioning", response_class=HTMLResponse)