Kalbe-x-Bangkit
commited on
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
|