Update app.py
Browse files
app.py
CHANGED
|
@@ -49,8 +49,11 @@ async def predict(soil_data: SoilInput):
|
|
| 49 |
input_df = pd.DataFrame([input_data])
|
| 50 |
|
| 51 |
# Ne asigurăm că ordinea coloanelor este corectă
|
| 52 |
-
feature_order = ['cement_perecent', 'curing_period', 'compaction_rate']
|
| 53 |
-
input_df = input_df[feature_order]
|
|
|
|
|
|
|
|
|
|
| 54 |
|
| 55 |
# Facem predicția
|
| 56 |
prediction = model.predict(input_df)
|
|
|
|
| 49 |
input_df = pd.DataFrame([input_data])
|
| 50 |
|
| 51 |
# Ne asigurăm că ordinea coloanelor este corectă
|
| 52 |
+
#feature_order = ['cement_perecent', 'curing_period', 'compaction_rate']
|
| 53 |
+
#input_df = input_df[feature_order]
|
| 54 |
+
|
| 55 |
+
expected_features = model.feature_names_in_ # Extrage ordinea corectă din model
|
| 56 |
+
input_df = input_df[expected_features] # Reordonează caracteristicile
|
| 57 |
|
| 58 |
# Facem predicția
|
| 59 |
prediction = model.predict(input_df)
|