Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -78,11 +78,11 @@ class SemanticSearch:
|
|
78 |
return neighbors
|
79 |
|
80 |
def get_text_embedding(self, texts, batch=1000):
|
81 |
-
|
82 |
-
|
83 |
-
|
84 |
-
|
85 |
-
|
86 |
embeddings = np.vstack(embeddings)
|
87 |
return embeddings
|
88 |
|
|
|
78 |
return neighbors
|
79 |
|
80 |
def get_text_embedding(self, texts, batch=1000):
|
81 |
+
embeddings = []
|
82 |
+
for i in range(0, len(texts), batch):
|
83 |
+
text_batch = texts[i:(i+batch)]
|
84 |
+
emb_batch = self.use(text_batch)
|
85 |
+
embeddings.append(emb_batch)
|
86 |
embeddings = np.vstack(embeddings)
|
87 |
return embeddings
|
88 |
|