Fix for cpu

scripts/td_abg.py

@@ -58,7 +58,7 @@ def rmbg_fn(img):
 
 def refinement(img, mask, fast, psp_L):
     mask =  cv2.cvtColor(mask, cv2.COLOR_RGB2GRAY)
-    refiner = refine.Refiner(device='cuda:0') # device can also be 'cpu'
+    refiner = refine.Refiner(device='cpu') # device can also be 'cpu'
 
     # Fast - Global step only.
     # Smaller L -> Less memory usage; faster in fast mode.
@@ -106,7 +106,7 @@ def get_foreground(img, td_abg_enabled, h_split, v_split, n_cluster, alpha, th_r
     if cascadePSP_enabled == True and td_abg_enabled == False:
         mask = get_mask(img)
         mask = (mask * 255).astype(np.uint8)
-        refiner = refine.Refiner(device='cuda:0')
+        refiner = refine.Refiner(device='cpu')
         mask = refiner.refine(img, mask, fast=fast, L=psp_L)
         img = np.dstack((img, mask))