Spaces:
Sleeping
Sleeping
[email protected]
commited on
Commit
·
091d337
1
Parent(s):
6d6f559
modified endpoint
Browse files
main.py
CHANGED
@@ -20,7 +20,7 @@ def get_dimension(message: str):
|
|
20 |
|
21 |
|
22 |
@app.post("/classify/")
|
23 |
-
def
|
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)
|