Spaces:
Sleeping
Sleeping
Commit
·
ac04d4d
1
Parent(s):
892a768
fix: issues
Browse files
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()
|
| 9 |
-
|
|
|
|
|
|
| 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
|