fffiloni commited on
Commit
0639262
1 Parent(s): 2cb0c9e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -2,13 +2,13 @@ import sys
2
  import os
3
  from pathlib import Path
4
 
5
- # Add the StableCascade directory to the Python path
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
 
2
  import os
3
  from pathlib import Path
4
 
5
+ # Add the StableCascade and CSD directories to the Python path
6
+ app_dir = Path(__file__).parent
7
+ sys.path.extend([
8
+ str(app_dir),
9
+ str(app_dir / "third_party" / "StableCascade"),
10
+ str(app_dir / "third_party" / "CSD")
11
+ ])
12
 
13
  import yaml
14
  import torch