Update app.py

app.py

@@ -21,7 +21,7 @@ import gdown
 import matplotlib.pyplot as plt
 import warnings
 warnings.filterwarnings("ignore")
-
+import requests
 os.system("git clone https://github.com/xuebinqin/DIS")
 os.system("mv DIS/IS-Net/* .")
 
@@ -53,7 +53,7 @@ class GOSNormalize(object):
 transform =  transforms.Compose([GOSNormalize([0.5,0.5,0.5],[1.0,1.0,1.0])])
 
 def load_image(im_path, hypar):
-    im = im_reader(im_path)
+    im = Image.open(requests.get(im_path, stream=True).raw)
     im, im_shp = im_preprocess(im, hypar["cache_size"])
     im = torch.divide(im,255.0)
     shape = torch.from_numpy(np.array(im_shp))
@@ -132,8 +132,7 @@ net = build_model(hypar, device)
 
 
 def inference(image):
-  image_path = "1.jpg"
-  gdown.download(image, image_path, quiet=False)
+  image_path = image
   image_tensor, orig_size = load_image(image_path, hypar)
   mask = predict(net, image_tensor, orig_size, hypar, device)