.

app.py

@@ -8,7 +8,7 @@ from PIL import Image, ImageDraw, ImageFont
 # Register the custom layer
 tf.keras.utils.get_custom_objects().update({'KerasLayer': hub.KerasLayer})
 
-# Now you can load the model without issues
+# Load the model
 model = tf.keras.models.load_model("dog_breed_model.h5")
 
 def load_breeds(file_path='unique_breeds.json'):
@@ -40,6 +40,11 @@ def predict_breed(image):
 
     # Draw the prediction and confidence over the image
     img_with_text = image.copy()
+    
+    # Convert to RGB mode if it's not
+    if img_with_text.mode != 'RGB':
+        img_with_text = img_with_text.convert('RGB')
+    
     draw = ImageDraw.Draw(img_with_text)
     
     # Set font for drawing text (ensure you have a suitable font installed)