Spaces:
Running
Running
Update app.py
Browse files
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 |
|