Jamshaid89 commited on
Commit
ddd1ddb
1 Parent(s): 7141b24

Removed debug

Browse files
Files changed (1) hide show
  1. app.py +0 -10
app.py CHANGED
@@ -20,20 +20,10 @@ base_model.trainable = False
20
 
21
  def create_model():
22
  input_layer = keras.Input(shape=(160,160,3))
23
- if debug:
24
- print(input_layer)
25
  R1 = Rescaling(scale=1/255)(input_layer)
26
- if debug:
27
- print(R1)
28
  B = base_model(R1 , training=False)
29
- if debug:
30
- print(B)
31
  P1 = keras.layers.GlobalAveragePooling2D()(B)
32
- if debug:
33
- print(P1)
34
  output_layer = Dense(1 , activation="sigmoid")(P1)
35
- if debug:
36
- print(f"output_layer: {output_layer}")
37
  model = keras.Model(input_layer , output_layer)
38
  model.compile(optimizer=keras.optimizers.Adam(0.001) , loss=keras.losses.BinaryCrossentropy() , metrics=["accuracy"])
39
  return model , input_layer, R1 , B , P1 , output_layer
 
20
 
21
  def create_model():
22
  input_layer = keras.Input(shape=(160,160,3))
 
 
23
  R1 = Rescaling(scale=1/255)(input_layer)
 
 
24
  B = base_model(R1 , training=False)
 
 
25
  P1 = keras.layers.GlobalAveragePooling2D()(B)
 
 
26
  output_layer = Dense(1 , activation="sigmoid")(P1)
 
 
27
  model = keras.Model(input_layer , output_layer)
28
  model.compile(optimizer=keras.optimizers.Adam(0.001) , loss=keras.losses.BinaryCrossentropy() , metrics=["accuracy"])
29
  return model , input_layer, R1 , B , P1 , output_layer