tding1 commited on
Commit
4d01905
·
verified ·
1 Parent(s): 82ba2a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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