Update app.py
Browse files
app.py
CHANGED
|
@@ -36,8 +36,8 @@ def preprocess_input(row_values):
|
|
| 36 |
input_df['byte_ratio'] = input_df['sbytes'] / (input_df['dbytes'] + 1)
|
| 37 |
input_df['pkt_ratio'] = input_df['Spkts'] / (input_df['Dpkts'] + 1)
|
| 38 |
|
| 39 |
-
#
|
| 40 |
-
input_df = input_df.drop(columns=features_to_drop + ['attack_cat', 'Label'], errors='ignore')
|
| 41 |
|
| 42 |
return input_df
|
| 43 |
|
|
|
|
| 36 |
input_df['byte_ratio'] = input_df['sbytes'] / (input_df['dbytes'] + 1)
|
| 37 |
input_df['pkt_ratio'] = input_df['Spkts'] / (input_df['Dpkts'] + 1)
|
| 38 |
|
| 39 |
+
# ✅ Fix: convert features_to_drop to list before adding with another list
|
| 40 |
+
input_df = input_df.drop(columns=list(features_to_drop) + ['attack_cat', 'Label'], errors='ignore')
|
| 41 |
|
| 42 |
return input_df
|
| 43 |
|