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
|