Spaces:
Runtime error
Runtime error
Jamshaid89
commited on
Commit
•
ddd1ddb
1
Parent(s):
7141b24
Removed debug
Browse files
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
|