Solar-Iz commited on
Commit
f73a5f3
·
1 Parent(s): 27eaf67

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -4,12 +4,15 @@ import shap
4
  from catboost import CatBoostRegressor
5
  import matplotlib.pyplot as plt
6
 
7
- # Загрузка модели
8
- final_model = CatBoostRegressor()
9
- final_model.load_model('Model/best_model.cbm')
 
 
 
 
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("Прогнозирование цены автомобиля")