Update app.py

app.py

@@ -17,6 +17,8 @@ def greet(image):
     print(type(image))
     image = torch.tensor(image).float()
     print(image.min(), image.max())
+    image = image/255.0
+    image = torch.permute(image, [0,2,3,1])
     out = model(image)
     # model.train()
     return f"Hello {model.bn1.running_mean.data} !!"