EdBoy2202 commited on
Commit
0adf401
·
verified ·
1 Parent(s): 11ed7a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -7
app.py CHANGED
@@ -152,19 +152,13 @@ if st.session_state.image is not None:
152
  st.write(f"Identified Car Make: {make_name}")
153
  st.write(f"Identified Car Model: {model_name}")
154
 
155
- # Get additional information using GPT-3.5-turbo
156
- current_year = datetime.now().year
157
- overview = get_car_overview(make_name, model_name, current_year)
158
- st.write("Car Overview:")
159
- st.write(overview)
160
-
161
  # Find the closest match in the CSV based on the classification
162
  car_data = load_car_data()
163
  if car_data is not None:
164
  processed_car_data, label_encoders = preprocess_car_data(car_data)
165
  closest_car = find_closest_car(processed_car_data, label_encoders, make_name, model_name, current_year)
166
 
167
- st.write(f"Closest match in database:")
168
  st.write(f"Year: {closest_car['year']}")
169
  st.write(f"Make: {label_encoders['make'].inverse_transform([closest_car['make']])[0]}")
170
  st.write(f"Model: {label_encoders['model'].inverse_transform([closest_car['model']])[0]}")
@@ -174,6 +168,14 @@ if st.session_state.image is not None:
174
  st.write(f"Transmission: {closest_car['transmission']}")
175
  st.write(f"Drive: {closest_car['drive']}")
176
  st.write(f"Type: {closest_car['type']}")
 
 
 
 
 
 
 
 
177
  else:
178
  st.error("Could not classify the image. Please try again with a different image.")
179
  else:
 
152
  st.write(f"Identified Car Make: {make_name}")
153
  st.write(f"Identified Car Model: {model_name}")
154
 
 
 
 
 
 
 
155
  # Find the closest match in the CSV based on the classification
156
  car_data = load_car_data()
157
  if car_data is not None:
158
  processed_car_data, label_encoders = preprocess_car_data(car_data)
159
  closest_car = find_closest_car(processed_car_data, label_encoders, make_name, model_name, current_year)
160
 
161
+ # st.write(f"Closest match in database:")
162
  st.write(f"Year: {closest_car['year']}")
163
  st.write(f"Make: {label_encoders['make'].inverse_transform([closest_car['make']])[0]}")
164
  st.write(f"Model: {label_encoders['model'].inverse_transform([closest_car['model']])[0]}")
 
168
  st.write(f"Transmission: {closest_car['transmission']}")
169
  st.write(f"Drive: {closest_car['drive']}")
170
  st.write(f"Type: {closest_car['type']}")
171
+
172
+ st.divider()
173
+
174
+ # Get additional information using GPT-3.5-turbo
175
+ current_year = datetime.now().year
176
+ overview = get_car_overview(make_name, model_name, current_year)
177
+ st.write("Car Overview:")
178
+ st.write(overview)
179
  else:
180
  st.error("Could not classify the image. Please try again with a different image.")
181
  else: