Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,12 +4,15 @@ import shap
|
|
| 4 |
from catboost import CatBoostRegressor
|
| 5 |
import matplotlib.pyplot as plt
|
| 6 |
|
| 7 |
-
# Загрузка модели
|
| 8 |
-
|
| 9 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
|
| 11 |
-
|
| 12 |
-
df = pd.read_csv('Dataset/car_data.txt', sep=',')
|
| 13 |
|
| 14 |
# Заголовок приложения
|
| 15 |
st.title("Прогнозирование цены автомобиля")
|
|
|
|
| 4 |
from catboost import CatBoostRegressor
|
| 5 |
import matplotlib.pyplot as plt
|
| 6 |
|
| 7 |
+
# Загрузка данных и модели
|
| 8 |
+
@st.cache(allow_output_mutation=True)
|
| 9 |
+
def load_data_and_model():
|
| 10 |
+
df = pd.read_csv('Dataset/car_data.txt', sep=',')
|
| 11 |
+
final_model = CatBoostRegressor()
|
| 12 |
+
final_model.load_model('Model/best_model.cbm')
|
| 13 |
+
return df, final_model
|
| 14 |
|
| 15 |
+
df, final_model = load_data_and_model()
|
|
|
|
| 16 |
|
| 17 |
# Заголовок приложения
|
| 18 |
st.title("Прогнозирование цены автомобиля")
|