RosiYo commited on
Commit
df16f8c
·
1 Parent(s): 4930e66

feat[bert]: FastAPI interface

Browse files
Files changed (1) hide show
  1. ml-service/app.py +2 -2
ml-service/app.py CHANGED
@@ -11,8 +11,8 @@ logging.basicConfig(level = logging.INFO)
11
  app = FastAPI()
12
  classifier = Classifier()
13
 
14
- @app.post("/sentiment")
15
- async def data(data: dict):
16
  try:
17
  input_text = data["text"]
18
  res = nlp.sentiment_analysis(classifier, input_text)
 
11
  app = FastAPI()
12
  classifier = Classifier()
13
 
14
+ @app.post("/predict")
15
+ def predict(data: dict):
16
  try:
17
  input_text = data["text"]
18
  res = nlp.sentiment_analysis(classifier, input_text)