Update app.py
Browse files
app.py
CHANGED
|
@@ -9,12 +9,12 @@ import requests
|
|
| 9 |
from tqdm import tqdm
|
| 10 |
import subprocess
|
| 11 |
|
| 12 |
-
# Clone
|
| 13 |
if not os.path.exists('InSPyReNet'):
|
| 14 |
print("Cloning InSPyReNet repository...")
|
| 15 |
subprocess.run(['git', 'clone', 'https://github.com/plemeri/InSPyReNet.git'])
|
| 16 |
|
| 17 |
-
#
|
| 18 |
sys.path.insert(0, os.path.abspath('InSPyReNet'))
|
| 19 |
sys.path.insert(0, os.path.abspath('InSPyReNet/lib'))
|
| 20 |
|
|
@@ -41,7 +41,7 @@ if not os.path.exists('InSPyReNet.pth'):
|
|
| 41 |
"InSPyReNet.pth"
|
| 42 |
)
|
| 43 |
|
| 44 |
-
# Import after setting
|
| 45 |
from InSPyReNet import InSPyReNet
|
| 46 |
from modules.layers import load_model
|
| 47 |
from utils.misc import load_config
|
|
|
|
| 9 |
from tqdm import tqdm
|
| 10 |
import subprocess
|
| 11 |
|
| 12 |
+
# Clone repository if not present
|
| 13 |
if not os.path.exists('InSPyReNet'):
|
| 14 |
print("Cloning InSPyReNet repository...")
|
| 15 |
subprocess.run(['git', 'clone', 'https://github.com/plemeri/InSPyReNet.git'])
|
| 16 |
|
| 17 |
+
# Set up correct Python paths
|
| 18 |
sys.path.insert(0, os.path.abspath('InSPyReNet'))
|
| 19 |
sys.path.insert(0, os.path.abspath('InSPyReNet/lib'))
|
| 20 |
|
|
|
|
| 41 |
"InSPyReNet.pth"
|
| 42 |
)
|
| 43 |
|
| 44 |
+
# Import after setting up environment
|
| 45 |
from InSPyReNet import InSPyReNet
|
| 46 |
from modules.layers import load_model
|
| 47 |
from utils.misc import load_config
|