Update app.py

app.py

@@ -6,12 +6,12 @@ import faiss
 import numpy as np
 import wget
 from PIL import Image
-from io import BytesIO
+# import io
 from sentence_transformers import SentenceTransformer
 import json
 from zipfile import ZipFile
 import zipfile
-from io import BytesIO
+# from io import BytesIO
 from PIL import Image
 # from huggingface_hub import hf_hub_download
 # hf_hub_download(repo_id="shivangibithel/Flickr8k", filename="Images.zip")
@@ -70,7 +70,8 @@ def search(query, k=5):
         text_id = i
         image_id = str(image_list[i])
         image_data = zip_file.read(image_id)
-        image = Image.open(io.BytesIO(image_data))
+        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)