soiz1 commited on
Commit
1b2575b
·
verified ·
1 Parent(s): 7bce9b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -12,6 +12,13 @@ from werkzeug.utils import secure_filename
12
  import warnings
13
  warnings.filterwarnings("ignore")
14
 
 
 
 
 
 
 
 
15
  # モデルと設定の初期化
16
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
17
 
 
12
  import warnings
13
  warnings.filterwarnings("ignore")
14
 
15
+ os.system("git clone https://github.com/xuebinqin/DIS")
16
+ os.system("mv DIS/IS-Net/* .")
17
+
18
+ # project imports
19
+ from data_loader_cache import normalize, im_reader, im_preprocess
20
+ from models import *
21
+
22
  # モデルと設定の初期化
23
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
24