bteodoru commited on
Commit
b70abde
·
verified ·
1 Parent(s): 70aa361

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -38,7 +38,8 @@ except Exception as e:
38
  print(f"Eroare la încărcarea modelului: {str(e)}")
39
  raise
40
 
41
- FEATURE_ORDER = ['cement_perecent', 'curing_period', 'compaction_rate']
 
42
  @app.post("/predict")
43
  async def predict(soil_data: SoilInput):
44
  """
 
38
  print(f"Eroare la încărcarea modelului: {str(e)}")
39
  raise
40
 
41
+ # FEATURE_ORDER = ['cement_perecent', 'curing_period', 'compaction_rate']
42
+ FEATURE_ORDER = model.feature_names_in_
43
  @app.post("/predict")
44
  async def predict(soil_data: SoilInput):
45
  """