Update app.py

app.py

@@ -27,7 +27,7 @@ def create_retrieval_figure(res):
     for i, image in zip(range(len(res)), res):
         if i >= 10:
             break
-        current_image_path = image.split("/")[3]+"/"+image.split("/")[4]
+        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)
@@ -35,7 +35,7 @@ def create_retrieval_figure(res):
         ax = fig.add_subplot(rows, cols, i + 1)
         plt.axis('off')
         plt.imshow(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
-        item_uuid = current_image_path.split("/")[1].split("_photoUUID")[0].split("itemUUID_")[1]
+        item_uuid = current_image_path.split("/")[2].split("_photoUUID")[0].split("itemUUID_")[1]
         ax.set_title('Top {}'.format(i), fontsize=40)
         names = names + "Top " + str(i) + " item UUID is " + item_uuid + "\n"
     return fig, names