Update bg.py

bg.py

@@ -14,6 +14,8 @@ import tempfile
 
 warnings.filterwarnings("ignore")
 
+os.system("git clone https://github.com/xuebinqin/DIS")
+os.system("mv DIS/IS-Net/* .")
 
 # project imports
 from data_loader_cache import normalize, im_reader, im_preprocess 
@@ -22,6 +24,11 @@ from models import *
 #Helpers
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
+# Download official weights
+if not os.path.exists("saved_models"):
+    os.mkdir("saved_models")
+    MODEL_PATH_URL = "https://drive.google.com/uc?id=1KyMpRjewZdyYfxHPYcd-ZbanIXtin0Sn"
+    gdown.download(MODEL_PATH_URL, "saved_models/isnet.pth", use_cookies=False)
 
 class GOSNormalize(object):
     '''