Update app.py

app.py

@@ -14,6 +14,8 @@ model = load_model()
 
 # Preprocessing function
 def preprocess_image(image):
+    # Convert to RGB to ensure 3 channels
+    image = image.convert("RGB")
     image = image.resize((224, 224))
     image_array = np.array(image) / 255.0
     image_array = np.expand_dims(image_array, axis=0)  # Add batch dimension