Update app.py

app.py

@@ -21,12 +21,6 @@ model_name = "sentence-transformers/all-distilroberta-v1"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = SentenceTransformer(model_name)
 
-# # Load the FAISS index
-# index_name = 'index.faiss'
-# index_url = './'
-# wget.download(index_url, index_name)
-# index = faiss.read_index(faiss_flickr8k.index)
-
 # Define the path to the zip folder containing the images
 zip_path = "Images.zip"
 
@@ -63,14 +57,10 @@ def search(query, k=5):
     for i in I[0]:
         text_id = i
         image_id = str(image_list[i])
-        image_data = zip_file.read(image_id)
+        image_data = zip_file.read("Images/" +image_id)
         image = Image.open(image_data)
-        # image = Image.open(io.BytesIO(image_data))
         st.image(image, caption=image_name, width=200)
-        # image_url = "./Images/" + image_id
-        # image_urls.append(image_url)
 
-    # return image_urls
 
 st.title("Image Search App")