Update app.py

app.py

@@ -103,9 +103,9 @@ st.sidebar.title("Options")
 training_mode = st.sidebar.radio("Select training mode", ["None", "Dropout", "Batch Normalization"])
 
 if training_mode == 'None':
-    user_choice == 'Predict'
+    user_choice = 'Predict'
 elif training_mode == "Dropout" or "Batch Normalization":
-    user_choice == 'Train'
+    user_choice = 'Train'
 
 
 # Display the current training mode and user choice