Upload app.py

app.py

@@ -169,22 +169,17 @@ def search(text_query, image_query, top_k: int = 10):
     #     {"Score": [torch.round(torch.tensor(value)*100)/100 for value in top_k_scores], "Title": top_k_titles, "Text": top_k_texts}
     # )
     # 获取图像
-    image_output = []
-    for img in info:
-        with zipfile.ZipFile('./image.zip', 'r') as zip_ref:
-        # 读取图像文件
-            with zip_ref.open(imgs_folder + img.replace('_','/')) as image_file:
-                # 将读取的字节流转换为图像
-                image = Image.open(BytesIO(image_file.read()))
-                image_output.append(image)
+
 
     if text_query != None:
         # image_output = [Image.open(imgs_folder + img.replace('_','/')) for img in info]
         image_output = []
-        with zipfile.ZipFile('./image.zip', 'r') as zip_ref:
-            # 读取图像文件
-            for img in info:
-                with zip_ref.open(imgs_folder + img.replace('_', '/')) as image_file:
+        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)