Update app.py

app.py

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