Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -29,7 +29,7 @@ def TxtEmbed(text):
|
|
| 29 |
text_embeddings = F.layer_norm(text_embeddings, normalized_shape=(text_embeddings.shape[1],))
|
| 30 |
text_embeddings = F.normalize(text_embeddings, p=2, dim=1)
|
| 31 |
|
| 32 |
-
return text_embeddings.tolist();
|
| 33 |
|
| 34 |
|
| 35 |
|
|
|
|
| 29 |
text_embeddings = F.layer_norm(text_embeddings, normalized_shape=(text_embeddings.shape[1],))
|
| 30 |
text_embeddings = F.normalize(text_embeddings, p=2, dim=1)
|
| 31 |
|
| 32 |
+
return (text_embeddings.tolist())[0];
|
| 33 |
|
| 34 |
|
| 35 |
|