Update app.py

app.py

@@ -114,6 +114,10 @@ async def predict_catdog(file: UploadFile = File(...)):
     try:
         contents = await file.read()
         image = Image.open(io.BytesIO(contents))
+
+        if image.mode != 'RGB':
+            image = image.convert('RGB')
+                    
         _, height, width, _ = cat_dog_model.input_shape
         img_resized = image.resize((width, height))
         if img_resized.mode == 'RGBA':