Update main.py

main.py

@@ -60,6 +60,10 @@ def generate(item: Item):
 async def root():
     return {"status": "ok"}
 
+@app.head("/")
+async def root():
+    return {"status": "ok"}
+
 
 @app.post("/generate")
 async def generate_text(item: Item, token: Annotated[str, Depends(oauth2_scheme)]):