Spaces:

osimeoni
/

FOUND

Runtime error

osimeoni commited on Apr 7, 2023

Commit

bbefb98

1 Parent(s): 853c722

image shape

Files changed (1) hide show

app.py CHANGED Viewed

@@ -73,8 +73,8 @@ def predict(img_input):
     preds_up = (
         (sigmoid(preds_up.detach()) > 0.5).squeeze(0).float()
     )
-    return blend_images(img_pil, T.ToPILImage()(preds_up))
 title = 'FOUND - unsupervised object localization'

     preds_up = (
         (sigmoid(preds_up.detach()) > 0.5).squeeze(0).float()
     )
+    return blend_images(img_pil.resize([img_t.shape[-1], img_t.shape[-2]]), T.ToPILImage()(preds_up))
 title = 'FOUND - unsupervised object localization'