Update app.py

app.py

@@ -43,17 +43,17 @@ def greet(image):
         # image = np.transpose(image,(0,3,1,2))
         
         
-        # image = torch.tensor(image).float()
-        # image = image/255.0
-        # image = image.unsqueeze(0)
-        # image = torch.permute(image, [0,3,1,2])
+        image = torch.tensor(image).float()
+        image = image/255.0
+        image = image.unsqueeze(0)
+        image = torch.permute(image, [0,3,1,2])
 
         
         
         
         
-        transform_nor = transforms.Compose([transforms.ToTensor(), transforms.Resize((32,32)), transforms.Normalize((0.4914, 0.4822, 0.4465), (0.247, 0.243, 0.261))])
-        image = transform_nor(image).unsqueeze(0)
+        # transform_nor = transforms.Compose([transforms.ToTensor(), transforms.Resize((32,32)), transforms.Normalize((0.4914, 0.4822, 0.4465), (0.247, 0.243, 0.261))])
+        # image = transform_nor(image).unsqueeze(0)
         
         
         out = model(image)