Update main.py

main.py

@@ -56,6 +56,11 @@ def generate(item: Item):
         output += response.token.text
     return output
 
+@app.get("/")
+async def root():
+    return {"status": "ok"}
+
+
 @app.post("/generate/")
 async def generate_text(item: Item, token: Annotated[str, Depends(oauth2_scheme)]):
     # Reject if not authenticated