Update app.py

app.py

@@ -58,7 +58,8 @@ def predict(img) -> Tuple[Dict, float]:
     
     # Transform the target image and add a batch dimension
     image_transformed = test_transforms(np.array(img))
-    
+    transformed = test_transforms(image= image)
+    image_transformed = transformed["image"]
     image_transformed = image_transformed.unsqueeze(0)
     image_transformed = image_transformed.to(device)