Upload app.py

app.py

@@ -31,7 +31,12 @@ def load_model():
 
 model = load_model()
 
+class ConvertToRGB:
+    def __call__(self, img):
+        return img.convert("RGB")
+
 transform = transforms.Compose([
+    ConvertToRGB(),
     transforms.Resize((224, 224)),
     transforms.ToTensor(),
     transforms.Normalize(