Update app.py

app.py

@@ -12,7 +12,10 @@ transform = transforms.Compose([
     transforms.Resize(256),
     transforms.CenterCrop(224),
     transforms.ToTensor(),
-    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
+    transforms.Normalize(
+        mean=[0.485, 0.456, 0.406],
+        std=[0.229, 0.224, 0.225]
+    ),
 ])
 
 # Define the class labels
@@ -20,6 +23,9 @@ class_labels = ['Normal', 'Pneumonia']
 
 # Create a function to make predictions
 def predict(image):
+    # Convert the image to RGB
+    image = image.convert('RGB')
+    
     # Preprocess the image
     image = transform(image).unsqueeze(0)