Update app.py

app.py

@@ -39,13 +39,22 @@ best_softmax_reg.fit(small_x_train_reshape, small_y_train)
 
 best_nn = MLPClassifier(hidden_layer_sizes=(100, 100), activation='relu', solver='lbfgs')
 best_nn.fit(small_x_train_reshape, small_y_train)
-
+'''
 input_module1 = gr.Image(label = "test_image", image_mode='L', shape = (28,28))
 input_module2 = gr.Dropdown(choices=['Softmax Regression', 'Neural Network (sklearn)'], label = "Select Algorithm")
 output_module1 = gr.Textbox(label = "Predicted Class")
 output_module2 = gr.Label(label = "Predicted Probability")
+'''
+input_module1 = gr.Image(label = "test_image", image_mode='L')
+
+# Specify method dropdown menu
+input_module2 = gr.Dropdown(choices=['Softmax Regression', 'Neural Network (sklearn)', 'Neural Network (keras) two-layer','Neural Network (keras) three-layer'], label = "Select Algorithm")
 
+# Specify output 1
+output_module1 = gr.Textbox(label = "Predicted Class")
 
+# Specify output 2
+output_module2 = gr.Label(label = "Predict Probability")