Spaces:
Running
on
A100
Running
on
A100
Update app.py
Browse files
app.py
CHANGED
@@ -6,6 +6,10 @@ from pathlib import Path
|
|
6 |
stable_cascade_path = Path(__file__).parent / "third_party" / "StableCascade"
|
7 |
sys.path.append(str(stable_cascade_path))
|
8 |
|
|
|
|
|
|
|
|
|
9 |
import yaml
|
10 |
import torch
|
11 |
from tqdm import tqdm
|
|
|
6 |
stable_cascade_path = Path(__file__).parent / "third_party" / "StableCascade"
|
7 |
sys.path.append(str(stable_cascade_path))
|
8 |
|
9 |
+
# Add the CSD directory to the Python path
|
10 |
+
csd_path = Path(__file__).parent / "third_party" / "CSD"
|
11 |
+
sys.path.append(str(csd_path))
|
12 |
+
|
13 |
import yaml
|
14 |
import torch
|
15 |
from tqdm import tqdm
|