bteodoru commited on
Commit
34b730a
·
verified ·
1 Parent(s): b70abde

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -33,13 +33,14 @@ app = FastAPI()
33
  # Încărcăm modelul Random Forest direct
34
  try:
35
  model = joblib.load('rf_model.joblib')
 
36
  print("Model încărcat cu succes!")
37
  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
- FEATURE_ORDER = model.feature_names_in_
43
  @app.post("/predict")
44
  async def predict(soil_data: SoilInput):
45
  """
 
33
  # Încărcăm modelul Random Forest direct
34
  try:
35
  model = joblib.load('rf_model.joblib')
36
+ FEATURE_ORDER = model.feature_names_in_
37
  print("Model încărcat cu succes!")
38
  except Exception as e:
39
  print(f"Eroare la încărcarea modelului: {str(e)}")
40
  raise
41
 
42
  # FEATURE_ORDER = ['cement_perecent', 'curing_period', 'compaction_rate']
43
+
44
  @app.post("/predict")
45
  async def predict(soil_data: SoilInput):
46
  """