Update main.py

main.py

@@ -18,6 +18,10 @@ img_processor = AutoImageProcessor.from_pretrained(img_model_id)
 img_model = AutoModel.from_pretrained(img_model_id)
 img_model.eval()
 
+@app.get("/")
+def home():
+    return {"status": "online", "model": img_model_id, "endpoint": "/embed/image"}
+
 @app.post("/embed/image")
 async def embed_image(request: ImageRequest):
     try: