Updated the model_tsfm.

app.py

@@ -47,7 +47,7 @@ def predict(img) -> Tuple[Dict, float]:
     
     # Transform the target image and add a batch dimension
     #img = get_image(img_path, model_tsfm).unsqueeze(0)
-    img = model_tsfm(image=np.array(img))["image"]
+    img = model_tsfm(img)
     img = img.unsqueeze(0)
 
     # Put model into evaluation mode and turn on inference mode