Update app.py

app.py

@@ -15,12 +15,12 @@ class_names = joblib.load("class_names.pkl")
 
 # Manually defined feature names (from your dataset)
 FEATURE_COLUMNS = [
-    "flow_duration", "Header_Length", "Protocol Type", "Duration", "Rate", "Srate", "Drate",
+    "flow_duration", "header_length", "protocol_type", "duration", "rate", "srate", "drate",
     "fin_flag_number", "syn_flag_number", "rst_flag_number", "psh_flag_number", "ack_flag_number",
     "ece_flag_number", "cwr_flag_number", "ack_count", "syn_count", "fin_count", "urg_count",
-    "rst_count", "HTTP", "HTTPS", "DNS", "Telnet", "SMTP", "SSH", "IRC", "TCP", "UDP", "DHCP",
-    "ARP", "ICMP", "IPv", "LLC", "Tot sum", "Min", "Max", "AVG", "Std", "Tot size", "IAT",
-    "Number", "Magnitue", "Radius", "Covariance", "Variance", "Weight"
+    "rst_count", "http", "https", "dns", "telnet", "smtp", "ssh", "irc", "tcp", "udp", "dhcp",
+    "arp", "icmp", "ipv", "llc", "tot_sum", "min", "max", "avg", "std", "tot_size", "iat",
+    "number", "magnitue", "radius", "covariance", "variance", "weight"
 ]
 
 # Streamlit UI