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=10, validation_data=(x_val, y_val))  # Reduced epochs for quick testing
+    history = model.fit(x_train, y_train, epochs=100, validation_data=(x_val, y_val))  # Reduced epochs for quick testing
 
     # Save the model
     model.save("cifar10_cnn_model.h5")
@@ -53,7 +53,7 @@ else:
 class_names = ['airplane', 'automobile', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck']
 
 # Streamlit app title
-st.title("CIFAR-10 Image Classification")
+st.title("Image Detection System")
 
 # Upload image
 uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])