Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import streamlit as st
 import pickle
 import pandas as pd
@@ -41,12 +40,14 @@ input_data['island'] = island_encoder.transform(input_data['island'])
 input_data['sex'] = sex_encoder.transform(input_data['sex'])
 
 # Ensure the columns are in the correct order
-expected_columns = ['species', 'island', 'sex', 'bill_length_mm', 'bill_depth_mm', 'flipper_length_mm', 'body_mass_g']
-input_data = input_data[expected_columns]
+# Check if the model includes a ColumnTransformer and use it if available
+if isinstance(model, Pipeline):
+    preprocessor = model.named_steps.get('preprocessor')  # Replace with actual step name if different
+    if preprocessor:
+        input_data = preprocessor.transform(input_data)  # Apply any necessary transformations
 
 # Make prediction
 prediction = model.predict(input_data)
 
 # Show the result
 st.write(f'Predicted Species: {species_encoder.inverse_transform(prediction)}')
-