Update app.py
Browse files
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 |
-
|
| 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 |
"""
|