Update app.py

app.py

@@ -35,7 +35,7 @@ def greet(image):
         image = image/255.0
         image = image.unsqueeze(0)
         print(image.shape)
-        image = torch.permute(image, [0,2,3,1])
+        image = torch.permute(image, [0,3,1,2])
         out = model(image)
 
     # model.train()