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