Update app.py

app.py

@@ -128,18 +128,18 @@ def inference(image: 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(image).convert("RGB")
+  pil_mask = Image.fromarray(mask).convert("L")
+  im_rgb = Image.open(image).convert("RGB")
   
-  #im_rgba = im_rgb.copy()
-  #im_rgba.putalpha(pil_mask)
+  im_rgba = im_rgb.copy()
+  im_rgba.putalpha(pil_mask)
 
-  #return [im_rgba, pil_mask]
+  return [im_rgba, pil_mask]
 
-  image = (image_tensor.detach().cpu().numpy()*255).astype(np.uint8)
-  mask = (mask*255).astype(np.uint8)
+  #image = (image_tensor.detach().cpu().numpy()*255).astype(np.uint8)
+  #mask = (mask*255).astype(np.uint8)
 
-  return image, mask
+  #return image, mask
 
 title = "智能抠图"
 description = "可以从给定图像中删除背景的模型。 要使用它，只需上传您的图像<br>右键图片另存为，下载图片"