Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -52,6 +52,7 @@ def similarity_search(
|
|
| 52 |
|
| 53 |
if np.max(query_image) == 255:
|
| 54 |
query_image = query_image/255.
|
|
|
|
| 55 |
|
| 56 |
query_vector = feature_extractor.predict(
|
| 57 |
query_image[np.newaxis, ...], verbose=0)[0]
|
|
|
|
| 52 |
|
| 53 |
if np.max(query_image) == 255:
|
| 54 |
query_image = query_image/255.
|
| 55 |
+
query_image = tf.image.resize(query_image, (224, 224))
|
| 56 |
|
| 57 |
query_vector = feature_extractor.predict(
|
| 58 |
query_image[np.newaxis, ...], verbose=0)[0]
|