Update app/main.py

app/main.py

@@ -3,7 +3,7 @@ from pydantic import BaseModel
 from model.model import LLM
 import torch
 
-app = FastAPI()
+app = FastAPI(docs_url="/swagger-ui.html")
 
 class InputText(BaseModel):
     text: str
@@ -13,9 +13,17 @@ model_tag = "facebook/opt-125m"
 model = LLM(model_name = model_tag,
             device = "cuda" if torch.cuda.is_available() else "cpu")
 
+"""
 @app.get("/")
 async def root():
     return {"message": "Technical challenge OK"}
+"""
+
+@app.get("/")
+async def docs_redirect():
+    response = RedirectResponse(url='/swagger-ui.html')
+    return response
+
 
 @app.post("/language-detection")
 def language_detection(text):