Update app.py

app.py

@@ -46,6 +46,7 @@ def similarity_search(query_image, num_images=5, *_, feature_extractor=feature_e
 
     if np.max(query_image) == 255:
         query_image = query_image/255.
+    query_image = tf.image.resize(query_image, (224, 224))
 
     query_vector = feature_extractor.predict(
         query_image[np.newaxis, ...], verbose=0)[0]