feat[bert]: FastAPI interface

ml-service/app.py

@@ -4,30 +4,23 @@ import uvicorn
 from classifier import Classifier
 import nlp
 import logging
-from logging import log
 
 logging.basicConfig(level = logging.INFO)
 
 # Create the FastAPI instance
 app = FastAPI()
-router = APIRouter()
 classifier = Classifier()
 
 # Define the required routes
-# @router.get("/")
-# async def home():
-#   return {"message": "Machine Learning service"}
+@app.get("/")
+async def home():
+  return {"message": "Machine Learning service"}
 
-@router.post("/sentiment")
+@app.post("/sentiment")
 async def data(data: dict):
   try:
     input_text = data["text"]
     res = nlp.sentiment_analysis(classifier, input_text)
     return res
   except Exception as e:
-    logging.error("Something went wrong")
-
-app.include_router(router)
-
-if __name__ == "__main__":
-  uvicorn.run("app:app", reload=True, port=6000, host="0.0.0.0")
+    logging.error("Something went wrong")