Update main.py

main.py

@@ -7,7 +7,7 @@ import uvicorn
 import os
 from sklearn.preprocessing import StandardScaler
 import joblib
-
+from fastapi_swaggerui import get_swaggerui_html, get_swaggerui_settings
     
 """ Creating the FastAPI Instance. i.e. foundation for our API,
 which will be the main part of our project"""
@@ -77,7 +77,13 @@ def predict_sepsis(patient: Patient):
     scaled_data = scaler.transform(data)
     parsed = predict(df=scaled_data)
     return {"output": parsed}
-
+    
+# Serve Swagger UI
+app.get("/docs", include_in_schema=False)(get_swaggerui_html)
+app.get(
+    "/docs/openapi.json",
+    include_in_schema=False,
+)(lambda: get_swaggerui_settings(app, "/openapi.json", title="API"))
 
 if __name__ == "__main__":
     os.environ["DEBUG"] = "True"  # Enable debug mode