Update app.py

app.py

@@ -28,7 +28,7 @@ model.eval()
 
 def predict(im):
     
-    print(type(im))
+    print(im.shape)
 
     
     x = torch.tensor(im, dtype=torch.float32).unsqueeze(0).unsqueeze(0) / 255.