joyson commited on
Commit
a6e5854
·
verified ·
1 Parent(s): a422ab3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,13 +17,13 @@ def my_pipe(text):
17
  emb = np.array(vo.embed(text, model="voyage-2", input_type="document").embeddings)
18
 
19
  similarities = cosine_similarity(embeddings, emb)
20
- most_similar_indices = np.argsort(similarities, axis=0)[-5:]
21
 
22
  return [all_quotes[i] for i in most_similar_indices.flatten()]
23
 
24
  def launch(input):
25
  out = my_pipe(input)
26
- return out
27
 
28
  iface = gr.Interface(launch,
29
  inputs="text",
 
17
  emb = np.array(vo.embed(text, model="voyage-2", input_type="document").embeddings)
18
 
19
  similarities = cosine_similarity(embeddings, emb)
20
+ most_similar_indices = np.argsort(similarities, axis=0)[-3:]
21
 
22
  return [all_quotes[i] for i in most_similar_indices.flatten()]
23
 
24
  def launch(input):
25
  out = my_pipe(input)
26
+ return "\n".join(out)
27
 
28
  iface = gr.Interface(launch,
29
  inputs="text",