[email protected] commited on
Commit
091d337
·
1 Parent(s): 6d6f559

modified endpoint

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -20,7 +20,7 @@ def get_dimension(message: str):
20
 
21
 
22
  @app.post("/classify/")
23
- def get_dimension(message: str):
24
  loaded_model = joblib.load('sms_classifier_model.pkl')
25
  message_embedding = model.encode([message])
26
  prediction = loaded_model.predict(message_embedding)
 
20
 
21
 
22
  @app.post("/classify/")
23
+ def classify(message: str):
24
  loaded_model = joblib.load('sms_classifier_model.pkl')
25
  message_embedding = model.encode([message])
26
  prediction = loaded_model.predict(message_embedding)