Update app.py

app.py

@@ -2,7 +2,7 @@
 import gradio as gr
 
 # g. define Dropdown data type
-input_module1 = gr.inputs.Dropdown(choices=["KNN", "Softmax", "Deep Neural", "SVM", "Decision Tree","Random Forest","CNN"], label = "Model Selection")
+input_module1 = gr.inputs.Dropdown(choices=["KNN", "Softmax", "Deep Neural", "SVM", "Decision Tree","Random Forest"], label = "Model Selection")
 # b. define image data type
 input_module2 = gr.inputs.Dropdown(choices=["image1", "image2", "image3", "image4","image5"], label = "Sample Image")
 
@@ -81,12 +81,6 @@ def multi_inputs(input1, input2):
       loaded_model = pickle.load(open('randomforest.sav', 'rb'))
       image = img.reshape((1, -1))
       y_test_pred = loaded_model.predict(image)
-
-    if input1 == "CNN":
-      loaded_model = pickle.load(open('CNN.sav', 'rb'))
-      img = img.reshape((1, -1))
-      y_test_pred = loaded_model.predict(img)
-      y_test_pred = np.argmax(y_test_pred,axis=1)
     
     result = y_test_pred[0]