Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -64,6 +64,7 @@ def similarity_search(
|
|
| 64 |
|
| 65 |
if np.max(query_image) == 255:
|
| 66 |
query_image = query_image/255.
|
|
|
|
| 67 |
|
| 68 |
query_vector = feature_extractor.predict(
|
| 69 |
query_image[np.newaxis, ...], verbose=0)[0]
|
|
|
|
| 64 |
|
| 65 |
if np.max(query_image) == 255:
|
| 66 |
query_image = query_image/255.
|
| 67 |
+
query_image = tf.image.resize(query_image, (224, 224))
|
| 68 |
|
| 69 |
query_vector = feature_extractor.predict(
|
| 70 |
query_image[np.newaxis, ...], verbose=0)[0]
|