Spaces:
Sleeping
Sleeping
OmerFarooq
commited on
Commit
•
3618416
1
Parent(s):
f58cafb
Update app.py
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ def img_pros(img):
|
|
27 |
def create_model_mod(classes, activation):
|
28 |
inputs = keras.Input(shape = (160,160,3))
|
29 |
#normalizing pixel values
|
30 |
-
r = Rescaling(scale = 1
|
31 |
x = base_model(r, training = False)
|
32 |
gap = keras.layers.GlobalAveragePooling2D()(x)
|
33 |
outputs = keras.layers.Dense(classes ,activation = activation)(gap)
|
|
|
27 |
def create_model_mod(classes, activation):
|
28 |
inputs = keras.Input(shape = (160,160,3))
|
29 |
#normalizing pixel values
|
30 |
+
r = Rescaling(scale = 1./255)(inputs)
|
31 |
x = base_model(r, training = False)
|
32 |
gap = keras.layers.GlobalAveragePooling2D()(x)
|
33 |
outputs = keras.layers.Dense(classes ,activation = activation)(gap)
|