Input add resize.

app.py

@@ -348,7 +348,7 @@ def redirect_button(url):
 
 def predict(model, image):
     """ Predict bounding box and label for the input image. """
-    pred_bbox, pred_class = model.predict(enhancement_type)
+    pred_bbox, pred_class = model.predict(image)
     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
@@ -462,7 +462,10 @@ if uploaded_file is not None:
             if st.button('Auto Detect'):
                 st.write("Processing...")
                 # input_image = preprocess_image(image)
-                input_image = enhancement_type
+                # input_image = enhancement_type
+                input_image = cv2.resize(enhanced_image, (W, H))  # Resize the enhanced image to the required input size
+                input_image = (input_image - 127.5) / 127.5  # Normalize to [-1, +1]
+                input_image = np.expand_dims(input_image, axis=0).astype(np.float32)  # Expand dimensions and convert to float32
                 pred_bbox, pred_label, pred_label_confidence = predict(model, input_image)
         
                 # Updated label mapping based on the dataset