test update app.py for png

app.py

@@ -145,19 +145,18 @@ net = build_model(hypar, device)
 
 
 def inference(image):
-  image_path = image
-  image_tensor, orig_size = load_image(image_path, hypar)
-  mask = predict(net, image_tensor, orig_size, hypar, device)
+    image_path = image
+    image_tensor, orig_size = load_image(image_path, hypar)
+    mask = predict(net, image_tensor, orig_size, hypar, device)
 
-  pil_mask = Image.fromarray(mask).convert('L')
-  im_rgb = Image.open(get_url_im(image)).convert("RGB")
-  imrgba1 = im_rgb.copy()
-  imrgba2 = remove(imrgba1,210)
+    pil_mask = Image.fromarray(mask).convert('L')
+    im_rgb = Image.open(get_url_im(image)).convert("RGBA")
+    imrgba2 = remove(im_rgb.copy(), 210)
 
-  im_rgba = im_rgb.copy()
-  im_rgba.putalpha(pil_mask)
+    im_rgba = im_rgb.copy()
+    im_rgba.putalpha(pil_mask)
 
-  return im_rgba, imrgba2 , im_rgba
+    return im_rgba, imrgba2, im_rgba
 
 
 title = "Bg remover for SarvM catalog"