Update app.py

app.py

@@ -37,7 +37,7 @@ def predict(img) -> Tuple[Dict, float]:
     
     # Transform the target image and add a batch dimension
     img = effnetb2_transforms(img).unsqueeze(0)
-    pix = np.array(img.getdata()).reshape(img.size[0], img.size[1], 3)
+    pix = np.array(img)
     
     # Put model into evaluation mode and turn on inference mode
     effnetb2.eval()