Spaces:
Runtime error
Runtime error
Ezhil
commited on
Commit
·
bd08cec
1
Parent(s):
daacc73
forth commit
Browse files- main.py +1 -1
- services/sms_service.py +1 -1
main.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
from fastapi import FastAPI, HTTPException
|
| 2 |
-
from services.sms_service import predict_label, compute_cosine_similarity,
|
| 3 |
from schemas.input_schemas import CosineSimilarityInput, MessageInput, EmbeddingInput
|
| 4 |
|
| 5 |
app = FastAPI()
|
|
|
|
| 1 |
from fastapi import FastAPI, HTTPException
|
| 2 |
+
from services.sms_service import predict_label, compute_cosine_similarity, get_embedding
|
| 3 |
from schemas.input_schemas import CosineSimilarityInput, MessageInput, EmbeddingInput
|
| 4 |
|
| 5 |
app = FastAPI()
|
services/sms_service.py
CHANGED
|
@@ -47,7 +47,7 @@ async def compute_cosine_similarity(text1: str, text2: str):
|
|
| 47 |
except Exception as e:
|
| 48 |
raise HTTPException(status_code=500, detail=f"Error computing similarity: {str(e)}")
|
| 49 |
|
| 50 |
-
async def
|
| 51 |
try:
|
| 52 |
classifier, vectorizer = load_model()
|
| 53 |
|
|
|
|
| 47 |
except Exception as e:
|
| 48 |
raise HTTPException(status_code=500, detail=f"Error computing similarity: {str(e)}")
|
| 49 |
|
| 50 |
+
async def get_embedding(message: str):
|
| 51 |
try:
|
| 52 |
classifier, vectorizer = load_model()
|
| 53 |
|