Bashir Gulistani commited on
Commit
a857577
·
unverified ·
1 Parent(s): 7dbc07c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 /home/student-admin/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]):
 
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]):