Update app.py

app.py

@@ -39,7 +39,7 @@ if not os.path.exists("cifar10_cnn_model.h5"):
 
     # Train the model
     st.write("Training the model...")
-    history = model.fit(x_train, y_train, epochs=100, validation_data=(x_val, y_val))  # Reduced epochs for quick testing
+    history = model.fit(x_train, y_train, epochs=40, validation_data=(x_val, y_val))  # Reduced epochs for quick testing
 
     # Save the model
     model.save("cifar10_cnn_model.h5")