OmerFarooq commited on
Commit
f58cafb
1 Parent(s): 52fee21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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./255)(inputs)
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/127.5, offset = -1)(inputs)
31
  x = base_model(r, training = False)
32
  gap = keras.layers.GlobalAveragePooling2D()(x)
33
  outputs = keras.layers.Dense(classes ,activation = activation)(gap)