Update app.py

app.py

@@ -12,7 +12,8 @@ N_FEATURES = model.input_shape[1]
 
 def predict_csv(file):
     df = pd.read_csv(file)
-
+    # Drop label column
+    df = df.drop(columns=["Label", "label", "Class", "class"], errors="ignore")
     # Check number of features
     if df.shape[1] != N_FEATURES:
         return f"Expected {N_FEATURES} features, but got {df.shape[1]} columns."