Update app.py

app.py

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