Update Sniffer_AI.py

Sniffer_AI.py

@@ -26,13 +26,15 @@ class_labels = {
     9: "mitm"
 }
 
-def detect_intrusion(features, model_choice="Decision Tree"):
+def detect_intrusion(features, model_choice="Random Forest"):
     # 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 == "Decision Tree":
+    if model_choice == "Random Forest":
         model = rf_model
+    elif model_choice == "Decision Tree":
+        model = decision_tree_model
     elif model_choice == "Bagging Classifier":
         model = model_bagging
     elif model_choice == "AdaBoost Classifier":