Spaces:
Sleeping
Sleeping
Commit
·
cba38f7
1
Parent(s):
ac04d4d
fix: changes in model
Browse files
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 |
-
|
| 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
|
|
|