Upload app.py

app.py

@@ -85,7 +85,7 @@ def single_scale_test(image):
 #        fg_name = im_name.split('.')[0] + '_fg.png'
     Image.fromarray(((foreground).astype('uint8')), mode='RGB').save(os.path.join('MODNet/output-img', 'fg_name.png'))
     output = Image.open(os.path.join('MODNet/output-img', 'fg_name.png'))
-    image = np.array(output)
+    image = np.asarray(output)
 
     model = RCF().cuda()
     checkpoint = torch.load("RCFPyTorch0/bsds500_pascal_model.pth")
@@ -102,7 +102,7 @@ def single_scale_test(image):
     for i in range(len(results)):
         all_res[i, 0, :, :] = results[i]
     #filename = osp.splitext(test_list[idx])[0]
-    torchvision.utils.save_image(1 - all_res, osp.join('RCFPyTorch0/results/RCF', 'result.jpg'))
+    #torchvision.utils.save_image(1 - all_res, osp.join('RCFPyTorch0/results/RCF', 'result.jpg'))
     fuse_res = torch.squeeze(results[1].detach()).cpu().numpy()
     fuse_res = ((1 - fuse_res) * 255).astype(np.uint8)
     cv2.imwrite(osp.join("RCFPyTorch0/results/RCF", 'result_ss.png'), fuse_res)
@@ -143,4 +143,4 @@ os.environ['CUDA_VISIBLE_DEVICES'] = args.gpu
 
 
 interface = gr.Interface(fn=single_scale_test, inputs="image", outputs="image")
-interface.launch(share=True)
+interface.launch()