Kalbe-x-Bangkit commited on
Commit
aa253a3
·
verified ·
1 Parent(s): 67a2742

Change input image of detection into enhanced_image.

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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