Update Sniffer_AI.py

Sniffer_AI.py

@@ -26,19 +26,17 @@ class_labels = {
     9: "mitm"
 }
 
-def detect_intrusion(features, model_choice="Random Forest"):
+def detect_intrusion(features, model_choice="Decision Tree"):
     # Convert the input string (comma-separated values) into a list of floats
     features = [list(map(float, features.split(",")))]
     
     # Choose the model based on user selection
-    if model_choice == "Random Forest":
+    if model_choice == "Decision Tree":
         model = rf_model
-    elif model_choice == "Decision Tree":
-        model = dt_model
     elif model_choice == "Bagging Classifier":
-        model = bagging_model
+        model = model_bagging
     elif model_choice == "AdaBoost Classifier":
-        model = ada_model
+        model = model_adaboost
     else:
         return "Invalid model choice!"