Spaces:

srush
/

minichain

Running

srush commited on Mar 23, 2023

Commit

f14fdcf

1 Parent(s): abcc1d0

Upload with huggingface_hub

Files changed (1) hide show

gatsby.py CHANGED Viewed

@@ -24,7 +24,7 @@ gatsby.add_faiss_index("embeddings")
 @prompt(HuggingFaceEmbed("sentence-transformers/all-mpnet-base-v2"))
 def get_neighbors(model, inp, k=1):
     embedding = model(inp)
-    res = olympics.get_nearest_examples("embeddings", np.array(embedding), k)
     return res.examples["passages"]
 @prompt(OpenAI(),

 @prompt(HuggingFaceEmbed("sentence-transformers/all-mpnet-base-v2"))
 def get_neighbors(model, inp, k=1):
     embedding = model(inp)
+    res = gatsby.get_nearest_examples("embeddings", np.array(embedding), k)
     return res.examples["passages"]
 @prompt(OpenAI(),