Update app.py

app.py

@@ -39,8 +39,9 @@ def greet(image):
         # print(image.min(), image.max())
         # image = image/255.0
         transform_nor = transforms.Compose([transforms.ToTensor(), transforms.Normalize((0.4914, 0.4822, 0.4465), (0.247, 0.243, 0.261))])
-        
-        image = image.unsqueeze(0)
+
+        image = image[np.newaxis,:,:,:]
+        # image = image.unsqueeze(0)
         print(image.shape)
         image = torch.permute(image, [0,3,1,2])
         image = transform_nor(image)