Solar-Iz commited on
Commit
f288922
·
1 Parent(s): dac1eea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -15
app.py CHANGED
@@ -21,7 +21,6 @@ st.title("Прогнозирование цены автомобиля")
21
  inputs = {}
22
 
23
  # Выбор бренда
24
- unique_brands = df['brand'].unique()
25
  selected_brand = st.selectbox("Выберите бренд авто", unique_brands)
26
 
27
  # Выбор модели в зависимости от бренда
@@ -30,21 +29,10 @@ selected_model = st.selectbox("Выберите модель авто", brand_mo
30
  inputs['brand'] = selected_brand
31
  inputs['model'] = selected_model
32
 
33
- # Получаем категориальные столбцы из вашего DataFrame
34
- categorical_columns = ['brand', 'model', 'поколение', 'тип продавца', 'состояние',
35
- 'модификация', 'тип двигателя', 'коробка передач', 'привод',
36
- 'комплектация', 'тип кузова', 'цвет', 'авито оценка']
37
-
38
- numerical_columns = ['год выпуска', 'пробег', 'объем двигателя']
39
-
40
  # Остальные окошки для ввода параметров
41
- for column in df.columns:
42
- if column in categorical_columns:
43
- inputs[column] = st.selectbox(f"Выберите значение для {column}", df[column].unique())
44
- elif column in numerical_columns:
45
- inputs[column] = st.number_input(f"Введите значение для {column}")
46
- else:
47
- inputs[column] = st.text_input(f"Введите значение для {column}")
48
 
49
  # Кнопка для запуска предсказания
50
  if st.button("Предсказать цену"):
 
21
  inputs = {}
22
 
23
  # Выбор бренда
 
24
  selected_brand = st.selectbox("Выберите бренд авто", unique_brands)
25
 
26
  # Выбор модели в зависимости от бренда
 
29
  inputs['brand'] = selected_brand
30
  inputs['model'] = selected_model
31
 
 
 
 
 
 
 
 
32
  # Остальные окошки для ввода параметров
33
+ inputs['год выпуска'] = st.number_input("Введите значение для года выпуска")
34
+ inputs['пробег'] = st.number_input("Введите значение для пробега")
35
+ inputs['объем двигателя'] = st.number_input("Введите значение для объема двигателя")
 
 
 
 
36
 
37
  # Кнопка для запуска предсказания
38
  if st.button("Предсказать цену"):