v2

Signed-off-by: Dipankar Das <[email protected]>

app.py

@@ -1,9 +1,14 @@
 from fastapi import FastAPI
+from fastapi.responses import RedirectResponse
 from transformers import pipeline
  
 # Create a new FastAPI app instance
+
 app = FastAPI()
- 
+
+@app.get("/")
+async def docs_redirect():
+    return RedirectResponse(url='/docs')
 # Initialize the text generation pipeline
 # This function will be able to generate text
 # given an input.
@@ -14,9 +19,9 @@ model="google/flan-t5-small")
 # The generate() function is defined as a FastAPI route that takes a 
 # string parameter called text. The function generates text based on the # input using the pipeline() object, and returns a JSON response 
 # containing the generated text under the key "output"
-@app.get("/")
-async def root():
-    return {"message": "Hello World"}
+# @app.get("/")
+# async def root():
+#     return {"message": "Hello World"}
 
 @app.get("/generate")
 def generate(text: str):