Update app.py

app.py

@@ -16,11 +16,11 @@ model = SentenceTransformer(model_name)
 
 # # Load the FAISS index
 # index_name = 'index.faiss'
-# index_url = 'https://huggingface.co/spaces/shivangibithel/Text2ImageRetrieval/blob/main/faiss_flickr8k.index'
+# index_url = './'
 # wget.download(index_url, index_name)
-# index = faiss.read_index(index_name)
+# index = faiss.read_index(faiss_flickr8k.index)
 
-vectors = np.load("shivangibithel/Text2ImageRetrieval/sbert_text_features.npy")
+vectors = np.load("./sbert_text_features.npy")
 vector_dimension = vectors.shape[1]
 index = faiss.IndexFlatL2(vector_dimension)
 faiss.normalize_L2(vectors)
@@ -28,7 +28,7 @@ index.add(vectors)
 
 # Map the image ids to the corresponding image URLs
 image_map_name = 'captions.json'
-image_map_url = 'https://huggingface.co/spaces/shivangibithel/Text2ImageRetrieval/blob/main/captions.json'
+image_map_url = './captions.json'
 wget.download(image_map_url, image_map_name)
 
 with open(image_map_name, 'r') as f:
@@ -52,7 +52,7 @@ def search(query, k=5):
     for i in I[0]:
         text_id = i
         image_id = str(image_list[i])
-        image_url = "https://huggingface.co/spaces/shivangibithel/Text2ImageRetrieval/blob/main/Images/" + image_id
+        image_url = "./Images/" + image_id
         image_urls.append(image_url)
 
     return image_urls