Change input image of detection into enhanced_image.
Browse files
app.py
CHANGED
|
@@ -446,6 +446,7 @@ if uploaded_file is not None:
|
|
| 446 |
if st.button('Auto Detect'):
|
| 447 |
st.write("Processing...")
|
| 448 |
# input_image = preprocess_image(image)
|
|
|
|
| 449 |
pred_bbox, pred_label, pred_label_confidence = predict(model, input_image)
|
| 450 |
|
| 451 |
# Updated label mapping based on the dataset
|
|
|
|
| 446 |
if st.button('Auto Detect'):
|
| 447 |
st.write("Processing...")
|
| 448 |
# input_image = preprocess_image(image)
|
| 449 |
+
input_image = enhanced_image
|
| 450 |
pred_bbox, pred_label, pred_label_confidence = predict(model, input_image)
|
| 451 |
|
| 452 |
# Updated label mapping based on the dataset
|