Upload app.py

app.py

@@ -265,7 +265,17 @@ def img_search(image_query, top_k: int = 10):
     # )
     # 获取图像
     if text_query != None:
-        image_output = [Image.open(imgs_folder + img.replace('_','/')) for img in info]
+        # image_output = [Image.open(imgs_folder + img.replace('_','/')) for img in info]
+        image_output = []
+        for img in info:
+            sample_name = img.split('_')[0]
+            image_path = imgs_folder + sample_name + '.zip'
+            with zipfile.ZipFile(image_path, 'r') as zip_ref:
+                # 读取图像文件
+                with zip_ref.open(img.replace('_', '/')) as image_file:
+                    # 将读取的字节流转换为图像
+                    image = Image.open(BytesIO(image_file.read()))
+                    image_output.append(image)
     else:
         image_output = []
 
@@ -347,7 +357,17 @@ def txt_search(text_query, top_k: int = 10):
     # )
     # 获取图像
     if text_query != None:
-        image_output = [Image.open(imgs_folder + img.replace('_', '/')) for img in info]
+        # image_output = [Image.open(imgs_folder + img.replace('_','/')) for img in info]
+        image_output = []
+        for img in info:
+            sample_name = img.split('_')[0]
+            image_path = imgs_folder + sample_name + '.zip'
+            with zipfile.ZipFile(image_path, 'r') as zip_ref:
+                # 读取图像文件
+                with zip_ref.open(img.replace('_', '/')) as image_file:
+                    # 将读取的字节流转换为图像
+                    image = Image.open(BytesIO(image_file.read()))
+                    image_output.append(image)
     else:
         image_output = []