AaSiKu commited on
Commit
ac222f2
·
verified ·
1 Parent(s): 7d65595

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -7
app.py CHANGED
@@ -9,13 +9,7 @@ from keras.applications.inception_v3 import InceptionV3
9
  from keras.models import Model
10
 
11
  model_imagenet = InceptionV3(weights='imagenet',include_top=False,input_shape=(180, 180, 3))
12
-
13
- new_input = model.input
14
-
15
- hidden_layer = model.layers[-2].output
16
-
17
- model_new = Model(new_input, hidden_layer)
18
- model = Sequential()
19
  num_classes = 2
20
  data_aug_layer = tf.keras.Sequential([
21
  RandomFlip("horizontal"), # Rotate by up to 20 degrees
 
9
  from keras.models import Model
10
 
11
  model_imagenet = InceptionV3(weights='imagenet',include_top=False,input_shape=(180, 180, 3))
12
+ model_imagenet.trainable = False
 
 
 
 
 
 
13
  num_classes = 2
14
  data_aug_layer = tf.keras.Sequential([
15
  RandomFlip("horizontal"), # Rotate by up to 20 degrees