Update app.py

app.py

@@ -21,7 +21,7 @@ def create_model():
         tf.keras.layers.Dropout(0.5),
         tf.keras.layers.Dense(64, activation='relu'),
         tf.keras.layers.Dropout(0.5),
-        tf.keras.layers.Dense(10, activation='softmax')  # Adjust the number of classes accordingly
+        tf.keras.layers.Dense(40, activation='softmax')  # Adjust the number of classes to match the original model
     ])
     return model