Update app.py

app.py

@@ -16,7 +16,7 @@ from tensorflow.keras.models import load_model
 
 input_1 = gr.Image(shape=(28,28),image_mode='L')
  
-input_2 = gr.Dropdown(["SoftMax", "KNN", "Deep Neural Network", "SVM", "Decision Tree", "Random Forest"])
+input_2 = gr.Dropdown(["SoftMax", "KNN", "Deep Neural Network", "Decision Tree", "Random Forest"])
 
 output = gr.Label(num_top_classes=6)
 
@@ -35,11 +35,6 @@ def predict_neural(test_img):
   predictions = Neural_model.predict(test_img)
   return {i: float(predictions[0][i]) for i in range(0,10)}
   
-def predict_svm(test_img):
-  svm_model = pickle.load(open('svm_clf.pkl', 'rb'))
-  predictions = svm_model.predict_proba(test_img)
-  return {i: float(predictions[0][i]) for i in range(0,10)}
-  
 def predict_tree(test_img):
   tree_model = pickle.load(open('tree_clf.pkl', 'rb'))
   predictions = tree_model.predict_proba(test_img)