cogcorp commited on
Commit
97e95ab
·
1 Parent(s): 5198954

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -78,11 +78,11 @@ class SemanticSearch:
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
 
 
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