Change input of detection from enhanced_image to enhancement_type.

app.py

@@ -347,7 +347,7 @@ def redirect_button(url):
 
 def predict(model, image):
     """ Predict bounding box and label for the input image. """
-    pred_bbox, pred_class = model.predict(enhanced_image)
+    pred_bbox, pred_class = model.predict(enhancement_type)
     pred_label_confidence = np.max(pred_class, axis=1)[0]
     pred_label = np.argmax(pred_class, axis=1)[0]
     return pred_bbox[0], pred_label, pred_label_confidence
@@ -461,7 +461,7 @@ if uploaded_file is not None:
             if st.button('Auto Detect'):
                 st.write("Processing...")
                 # input_image = preprocess_image(image)
-                input_image = enhanced_image
+                input_image = enhancement_type
                 pred_bbox, pred_label, pred_label_confidence = predict(model, input_image)
         
                 # Updated label mapping based on the dataset