Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)[-
|
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",
|