rrg92 commited on
Commit
645dc98
·
verified ·
1 Parent(s): e2eb3d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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