Update app.py

app.py

@@ -3,10 +3,15 @@ import gradio as gr
 from datasets import load_dataset
 from transformers import AutoModel, AutoFeatureExtractor
 import wikipedia
+import numpy as np
 
 # Only runs once when the script is first run.
-with open("slugs_1024_new.pickle", "rb") as handle:
-    index = pickle.load(handle)
+#with open("slugs_1024_new.pickle", "rb") as handle:
+#    index = pickle.load(handle)
+
+embs= np.load('slugs_embeddings_1024k.npy', 'rb')
+index = pynndescent.NNDescent(embs, metric="cosine")
+index.prepare()
 
 # Load model for computing embeddings.
 feature_extractor = AutoFeatureExtractor.from_pretrained("sasha/autotrain-sea-slug-similarity-2498977005")