Update main.py

main.py

@@ -38,10 +38,10 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-
+@app.get("/", tags=["Home"])
 def api_home():
     return {'detail': 'Welcome to FastAPI TextGen Tutorial!'}
 
-
+@app.post("/api/generate", summary="Generate text from prompt", tags=["Generate"], response_model=Generate)
 def inference(input_prompt: str):
-    return generate_text(prompt=input_prompt)
+    return generate_text(prompt=input_prompt)