Update app.py

app.py

@@ -7,8 +7,8 @@ import pickle
 
 # st.set_page_config(layout="wide")
 
-st.title("Калькулятор оценки фракции выброса левого желудочка у пациентов с хронической сердечной недостаточностью на основании клинических и лабораторных параметров")
-
+st.title("Калькулятор прогнозирования значения фракции выброса левого желудочка")
+st.title("Кафедра пропедевтической терапии с курсом кардиологи")
 
 # col1, col2, col3, col4, col5, col6, col7, col8 = st.rows(8)
 
@@ -21,10 +21,7 @@ cd = 3
 nt = 394.05
 st2 = 30.8
 
-mod = st.radio(
-        "Выберите модель",
-        options=["Линейная", "Нелинейная"]
-)
+#mod = st.radio("Выберите модель", options=["Линейная", "Нелинейная"])
 
 # with col1:
 f1 = st.selectbox(
@@ -64,13 +61,13 @@ st2_check = st.checkbox('ST2', value=True)
 filename_model = 'class_model_logreg.pickle'
 filename_scaler = 'class_scaler.pickle'
 filename_ohe = 'class_one_hot_enc.pickle'
-filename_cat = 'class_model_catboost.pickle'
+#filename_cat = 'class_model_catboost.pickle'
 
 
 loaded_model = pickle.load(open(filename_model, 'rb'))
 loaded_scaler = pickle.load(open(filename_scaler, 'rb'))
 loaded_ohe = pickle.load(open(filename_ohe, 'rb'))
-loaded_cat = pickle.load(open(filename_cat, 'rb'))
+#loaded_cat = pickle.load(open(filename_cat, 'rb'))
 
 
 
@@ -176,10 +173,10 @@ if st.button('OK'):
 
     line_norm = loaded_scaler.transform(X_encode)
     
-    if mod == 'Линейная':
-        res = loaded_model.predict(line_norm)[0]
-        st.header(get_interval_by_class(int(res)))
-    else:
-        res = loaded_cat.predict(line_norm)[0]
-        st.header(get_interval_by_class(int(res)))
+    #if mod == 'Линейная':
+    #    res = loaded_model.predict(line_norm)[0]
+    #    st.header(get_interval_by_class(int(res)))
+    #else:
+    res = loaded_cat.predict(line_norm)[0]
+    st.header(get_interval_by_class(int(res)))