Spaces:
Running
Running
Commit
·
9bb3988
1
Parent(s):
90433fe
Update app.py
Browse files
app.py
CHANGED
@@ -104,7 +104,7 @@ training_mode = st.sidebar.radio("Select training mode", ["None", "Dropout", "Ba
|
|
104 |
|
105 |
if training_mode == 'None':
|
106 |
user_choice == 'Predict'
|
107 |
-
elif training_mode == "Dropout"
|
108 |
user_choice == 'Train'
|
109 |
|
110 |
|
|
|
104 |
|
105 |
if training_mode == 'None':
|
106 |
user_choice == 'Predict'
|
107 |
+
elif training_mode == "Dropout" or "Batch Normalization":
|
108 |
user_choice == 'Train'
|
109 |
|
110 |
|