Hugging Face's logo

Spaces:
tding1
/
p2p-demo
Sleeping

App Files Community
tding1 commited on
Commit
ad8163f
·
verified ·
1 Parent(s): ff3d1d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ class ImageSimilarity:
30
  self.batch_size = BATCH_SIZE
31
  self.img_size = IMG_SIZE
32
  self.img_dir = img_dir
33
- self.model = tf.keras.applications.EfficientNetB3(
34
  input_shape=(IMG_SIZE, IMG_SIZE, 3),
35
  # alpha=1.0,
36
  include_top=False,
 
30
  self.batch_size = BATCH_SIZE
31
  self.img_size = IMG_SIZE
32
  self.img_dir = img_dir
33
+ self.model = tf.keras.applications.EfficientNetB1(
34
  input_shape=(IMG_SIZE, IMG_SIZE, 3),
35
  # alpha=1.0,
36
  include_top=False,