Update Sniffer_AI.py

Sniffer_AI.py

@@ -1,11 +1,17 @@
 import gradio as gr
 import joblib
+import requests
+import os
+
+from sklearn.ensemble import RandomForestClassifier, BaggingClassifier, AdaBoostClassifier
+from sklearn.tree import DecisionTreeClassifier
+
 
 # Load the saved models
 rf_model = joblib.load('rf_model.pkl')
-dt_model = joblib.load('dt_model.pkl')
-bagging_model = joblib.load('bagging_model.pkl')
-ada_model = joblib.load('ada_model.pkl')
+dt_model = joblib.load('decision_tree_model.pkl')
+bagging_model = joblib.load('model_bagging.pkl')
+ada_model = joblib.load('model_adaboost.pkl')
 
 class_labels = {
     0: "normal",