convert input image to grayscale

app.py

@@ -11,7 +11,9 @@ from pathlib import Path
 # Function to perform image classification
 def classify_image(img):
     #imdata = np.asarray(Image.open(image_path))
-    alltransforms = torchvision.transforms.Compose([torchvision.transforms.ToTensor()])
+    alltransforms = torchvision.transforms.Compose([
+        torchvision.transforms.Grayscale(),
+        torchvision.transforms.ToTensor()])
     tensor_image = alltransforms(img)
     # bring it to the shape model expects N, C, H, W
     #print(tensor_image.shape)