Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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:
|