small

app.py

@@ -17,8 +17,8 @@ NORMALIZE = T.Normalize((0.485, 0.456, 0.406), (0.229, 0.224, 0.225))
 CACHE = True
 
 def blend_images(bg, fg, alpha=0.5):
-    fg = fg.convert('RGBA')
     bg = bg.convert('RGBA')
+    fg = fg.convert('RGBA')
     blended = Image.blend(bg, fg, alpha=alpha)
 
     return blended
@@ -67,7 +67,7 @@ def predict(img_input):
         (sigmoid(preds_up.detach()) > 0.5).squeeze(0).float()
     )
     
-    return blend_images(img_pil, preds_up)
+    return T.ToPILImage()(blend_images(img_pil, preds_up))
     
 
 title = 'FOUND'