initial commit

main.py

@@ -0,0 +1,27 @@
+from fastapi import FastAPI
+import joblib
+from sentence_transformers import SentenceTransformer
+
+app = FastAPI()
+model = SentenceTransformer(
+    'Alibaba-NLP/gte-base-en-v1.5', trust_remote_code=True)
+
+
+@app.get("/")
+def root():
+    return {"message": "Welcom to FastAPI with Logistic Regression"}
+
+
+@app.post("/dimensions/")
+def get_dimension(message: str):
+    message_embedding = model.encode([message])
+    return {"dimensions": message_embedding.shape[1]}
+
+
+@app.post("/classify/")
+def get_dimension(message: str):
+    loaded_model = joblib.load('sms_classifier_model.pkl')
+    message_embedding = model.encode([message])
+    prediction = loaded_model.predict(message_embedding)
+    return {"Predicted Category": f"{prediction[0]}"}
+

requirements.txt

@@ -0,0 +1,4 @@
+fastapi==0.115.5
+fastapi-cli==0.0.7
+joblib==1.4.2
+sentence-transformers==3.4.1

sms_classifier_model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d62c231572d4bb143da1a4fbe59ef0330779f3961331fbf3c854a92637509cd1
+size 7215