Solar-Iz commited on
Commit
5d62a07
·
1 Parent(s): 30a5ef1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -7
app.py CHANGED
@@ -10,10 +10,6 @@ final_model.load_model('Model/best_model.cbm')
10
  # Загрузка данных для предобработки
11
  df = pd.read_csv('Dataset/car_data.txt', sep=',')
12
 
13
- # Получаем уникальные бренды и их модели
14
- unique_brands = df['brand'].unique()
15
- brand_model_mapping = {brand: df[df['brand'] == brand]['model'].unique() for brand in unique_brands}
16
-
17
  # Заголовок приложения
18
  st.title("Прогнозирование цены автомобиля")
19
 
@@ -24,8 +20,8 @@ inputs = {}
24
  unique_brands = df['brand'].unique()
25
  selected_brand = st.selectbox("Выберите бренд авто", unique_brands)
26
 
27
- # Выбор модели в зависимости от бренда
28
- selected_model = st.selectbox("Выберите модель авто", df[df['brand'] == selected_brand]['model'].unique())
29
 
30
  inputs['brand'] = selected_brand
31
  inputs['model'] = selected_model
@@ -43,7 +39,7 @@ for column in df.columns:
43
  if column in numerical_columns:
44
  inputs[column] = st.number_input(f"Введите значение для {column}")
45
  elif column in categorical_columns:
46
- inputs[column] = st.selectbox(f"Выберите значение для {column}", df[column].unique())
47
 
48
  # Кнопка для запуска предсказания
49
  if st.button("Предсказать цену"):
 
10
  # Загрузка данных для предобработки
11
  df = pd.read_csv('Dataset/car_data.txt', sep=',')
12
 
 
 
 
 
13
  # Заголовок приложения
14
  st.title("Прогнозирование цены автомобиля")
15
 
 
20
  unique_brands = df['brand'].unique()
21
  selected_brand = st.selectbox("Выберите бренд авто", unique_brands)
22
 
23
+ # Выбор модели
24
+ selected_model = st.text_input("Введите модель авто")
25
 
26
  inputs['brand'] = selected_brand
27
  inputs['model'] = selected_model
 
39
  if column in numerical_columns:
40
  inputs[column] = st.number_input(f"Введите значение для {column}")
41
  elif column in categorical_columns:
42
+ inputs[column] = st.text_input(f"Введите значение для {column}")
43
 
44
  # Кнопка для запуска предсказания
45
  if st.button("Предсказать цену"):