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

fix: issues

Browse files
Files changed (1) hide show
  1. back_end/models/embedding_model.py +3 -2
back_end/models/embedding_model.py CHANGED
@@ -5,5 +5,6 @@ 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() # Convert NumPy array to list
9
- return embedding
 
 
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