Update app.py

app.py

@@ -52,12 +52,12 @@ def load_model_detection():
     )
     return model
 
-# def preprocess_image(image):
-#     """ Preprocess the image to the required size and normalization. """
-#     image = cv2.resize(image, (W, H))
-#     image = (image - 127.5) / 127.5  # Normalize to [-1, +1]
-#     image = np.expand_dims(image, axis=0).astype(np.float32)
-#     return image
+def preprocess_image(image):
+    """ Preprocess the image to the required size and normalization. """
+    image = cv2.resize(image, (W=320, H=320))
+    image = (image - 127.5) / 127.5  # Normalize to [-1, +1]
+    image = np.expand_dims(image, axis=0).astype(np.float32)
+    return image
 
 
 
@@ -467,7 +467,7 @@ if uploaded_file is not None:
         image = cv2.imdecode(file_bytes, 1) 
         if st.button('Auto Detect'):
             st.write("Processing...")
-            input_image = load_image(image)
+            input_image = preprocess_image(image)
             # 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]