Update app.py

app.py

@@ -29,9 +29,12 @@ def create_retrieval_figure(res):
             break
         current_image_path = "dataset/" + image.split("/")[3] + "/" + image.split("/")[4]
         archive = zipfile.ZipFile('dataset.zip', 'r')
-        imgfile = archive.read(current_image_path)
-        image = cv2.imdecode(np.frombuffer(imgfile, np.uint8), 1)
+        try:
+            imgfile = archive.read(current_image_path)
+            image = cv2.imdecode(np.frombuffer(imgfile, np.uint8), 1)
         # image_resized = cv2.resize(image, (224, 224), interpolation=cv2.INTER_LINEAR)
+        except Exception:
+            image = cv2.imdecode(np.array((3, 224, 224), np.uint8), 1)
         ax = fig.add_subplot(rows, cols, i + 1)
         plt.axis('off')
         plt.imshow(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))