Update app.py
Browse files
app.py
CHANGED
|
@@ -32,12 +32,12 @@ class ImageSimilarity:
|
|
| 32 |
self.img_dir = img_dir
|
| 33 |
self.model = tf.keras.applications.ResNet50V2(
|
| 34 |
input_shape=(IMG_SIZE, IMG_SIZE, 3),
|
| 35 |
-
alpha=1.0,
|
| 36 |
include_top=False,
|
| 37 |
weights="imagenet",
|
| 38 |
-
input_tensor=None,
|
| 39 |
-
pooling=None,
|
| 40 |
-
classifier_activation="softmax",
|
| 41 |
)
|
| 42 |
|
| 43 |
self.model.trainable = False
|
|
|
|
| 32 |
self.img_dir = img_dir
|
| 33 |
self.model = tf.keras.applications.ResNet50V2(
|
| 34 |
input_shape=(IMG_SIZE, IMG_SIZE, 3),
|
| 35 |
+
# alpha=1.0,
|
| 36 |
include_top=False,
|
| 37 |
weights="imagenet",
|
| 38 |
+
# input_tensor=None,
|
| 39 |
+
# pooling=None,
|
| 40 |
+
# classifier_activation="softmax",
|
| 41 |
)
|
| 42 |
|
| 43 |
self.model.trainable = False
|