Update app.py

app.py

@@ -17,7 +17,11 @@ class FixedDropout(tf.keras.layers.Layer):
     def call(self, inputs, training=None):
         if training is None:
             training = K.learning_phase()
-        return K.in_train_phase(K.dropout(inputs, self.rate, noise_shape=self.noise_shape, seed=self.seed), inputs, training=training)
+
+        if training:
+            return K.in_train_phase(K.dropout(inputs, self.rate, noise_shape=self.noise_shape, seed=self.seed), inputs, training=training)
+        else:
+            return inputs
 
     def get_config(self):
         config = super(FixedDropout, self).get_config()
@@ -26,6 +30,7 @@ class FixedDropout(tf.keras.layers.Layer):
         config['seed'] = self.seed  # Serialize the seed argument
         return config
 
+
 class ImageClassifierApp:
     def __init__(self, model_path):
         self.model_path = model_path