Ezhil commited on
Commit
bd08cec
·
1 Parent(s): daacc73

forth commit

Browse files
Files changed (2) hide show
  1. main.py +1 -1
  2. 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, get_embedding_from_transformer
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 compute_embeddings(message: str):
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