correct rgb match

app.py

@@ -16,13 +16,15 @@ model = load_model()
 transform = transforms.Compose([
     transforms.Resize((224, 224)),
     transforms.ToTensor(),
-    transforms.Normalize(mean=[0.5], std=[0.5])
+    transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])
 ])
 
 # Prediction function
 def predict(image):
+    if image is None:
+        return {"Error": 1.0}
     image = image.convert("RGB")
-    input_tensor = transform(image).unsqueeze(0)  # Add batch dim
+    input_tensor = transform(image).unsqueeze(0)
 
     with torch.no_grad():
         outputs = model(input_tensor)