Update src/train_model.py

src/train_model.py

@@ -125,7 +125,7 @@ def generate_flood_data(n: int = 5000):
 
     # Elevation
     elevation_features = spatial_join(base, elevation,
-                                      ["elevation_m", "slope_degrees",
+                                      ["elevation_m", 
                                        "flow_accumulation", "twi"])
     for col in elevation_features.columns:
         base[col] = elevation_features[col]