Bashir Gulistani
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -88,7 +88,7 @@ def clear_conversation():
|
|
88 |
#######
|
89 |
|
90 |
os.system("git clone https://github.com/xuebinqin/DIS")
|
91 |
-
os.system("mv DIS/IS-Net/* .")
|
92 |
|
93 |
from data_loader_cache import normalize, im_reader, im_preprocess
|
94 |
from models import *
|
@@ -97,7 +97,7 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
|
97 |
|
98 |
if not os.path.exists("saved_models"):
|
99 |
os.mkdir("saved_models")
|
100 |
-
os.system("mv /
|
101 |
|
102 |
class GOSNormalize(object):
|
103 |
def __init__(self, mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]):
|
|
|
88 |
#######
|
89 |
|
90 |
os.system("git clone https://github.com/xuebinqin/DIS")
|
91 |
+
os.system("mv products/DIS/IS-Net/* .")
|
92 |
|
93 |
from data_loader_cache import normalize, im_reader, im_preprocess
|
94 |
from models import *
|
|
|
97 |
|
98 |
if not os.path.exists("saved_models"):
|
99 |
os.mkdir("saved_models")
|
100 |
+
os.system("mv products/isnet.pth saved_models/")
|
101 |
|
102 |
class GOSNormalize(object):
|
103 |
def __init__(self, mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]):
|