bteodoru commited on
Commit
dd60af8
·
verified ·
1 Parent(s): d2cf429

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
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)