Mykes commited on
Commit
30364ee
·
verified ·
1 Parent(s): d48aa12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -8,12 +8,16 @@ import joblib
8
  # Загрузка сохраненной модели и других необходимых объектов
9
  @st.cache_resource
10
  def load_model():
11
- model = joblib.load('model.joblib')
 
12
  scaler = joblib.load('scaler.joblib')
13
  feature_list = joblib.load('features.joblib')
14
- return model, scaler, feature_list
 
 
 
15
 
16
- model, scaler, feature_list = load_model()
17
 
18
  st.title('Классификатор пациентов')
19
 
 
8
  # Загрузка сохраненной модели и других необходимых объектов
9
  @st.cache_resource
10
  def load_model():
11
+ with open('model.pkl', 'rb') as file:
12
+ model = pickle.load(file)
13
  scaler = joblib.load('scaler.joblib')
14
  feature_list = joblib.load('features.joblib')
15
+ categorical_features = joblib.load('categorical_features.joblib')
16
+ categorical_options = joblib.load('categorical_options.joblib')
17
+ group_names = joblib.load('group_names.joblib')
18
+ return model, scaler, feature_list, categorical_features, categorical_options, group_names
19
 
20
+ model, scaler, feature_list, categorical_features, categorical_options, group_names = load_model()
21
 
22
  st.title('Классификатор пациентов')
23