velmurugan1122 commited on
Commit
cba38f7
·
1 Parent(s): ac04d4d

fix: changes in model

Browse files
Files changed (1) hide show
  1. back_end/models/embedding_model.py +2 -3
back_end/models/embedding_model.py CHANGED
@@ -5,6 +5,5 @@ model = SentenceTransformer('Alibaba-NLP/gte-base-en-v1.5', trust_remote_code=Tr
5
 
6
  def generate_embedding(text: str):
7
  """Generate a 768-dimensional embedding for the input text."""
8
- embedding = model.encode(text).tolist()
9
- embedding_result = len(embedding)# Convert NumPy array to list
10
- return embedding_result
 
5
 
6
  def generate_embedding(text: str):
7
  """Generate a 768-dimensional embedding for the input text."""
8
+ embedding = model.encode(text).tolist() # Convert NumPy array to list
9
+ return embedding