Update app.py

app.py

@@ -39,7 +39,7 @@ def detect_objects(image, model_name):
     image_gray = image  # Use the image as is if already grayscale
   
   # Check if the model requires CNN specific preprocessing
-  if model_name in ["CNN_model", "CNN_with_reductions"]:
+  if model_name in ["CNN", "CNN_with_reductions"]:
     image_processed = np.array(image_gray)
     image_processed = image_processed.reshape(1, image_gray.shape[0], image_gray.shape[1], 1)
     image_processed = image_processed.astype('float32')