Update app.py

app.py

@@ -36,8 +36,10 @@ def find_similar(image):
     ## Return best image :)
     idx = sorted(zip(similarities, range(photo_features.shape[0])), key=lambda x: x[0], reverse=True)[0][1]
     photo_id = photo_ids[idx]
-    photo_data = photos[photos["photo_id"] == photo_id].iloc[0]
-
+    try:
+       photo_data = photos[photos["photo_id"] == photo_id].iloc[0]
+    except:
+       photo_data = photos.iloc[0]
     response = requests.get(photo_data["photo_image_url"] + "?w=640")
     img = PILIMAGE.open(BytesIO(response.content))
     return img