Debugging

Dockerfile

@@ -23,4 +23,5 @@ 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
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]
+#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

@@ -16,9 +16,13 @@ static_dir = Path(__file__).parent / "vit_captioning" / "static"
 app.mount("/static", StaticFiles(directory=static_dir), name="static")
 
 # ✅ Landing page at `/`
-@app.get("/", response_class=HTMLResponse)
-async def landing():
-    return Path("vit_captioning/static/landing.html").read_text()
+# @app.get("/", response_class=HTMLResponse)
+# async def landing():
+#     return Path("vit_captioning/static/landing.html").read_text()
+
+@app.get("/") # DEBUGGING
+def health_check():
+    return {"status": "ok"}
 
 # ✅ Captioning page at `/captioning`
 @app.get("/captioning", response_class=HTMLResponse)
@@ -47,5 +51,5 @@ async def generate(file: UploadFile = File(...)):
     captions = caption_generator.generate_caption(temp_file)
     return captions
 
-if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+# if __name__ == "__main__":
+#     uvicorn.run(app, host="0.0.0.0", port=8000)