Update app.py

app.py

@@ -37,6 +37,7 @@ def find_nearest_neighbors(encoded_images, input_image, top_n=5):
     return [(index, distances[index]) for index in nearest_neighbors]
 
 def get_image(index):
+    split = len(dataset["train"])
     if index < split:
         return dataset["train"][index]
     else: