Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -23,8 +23,7 @@ class Examples(gr.helpers.Examples):
|
|
| 23 |
|
| 24 |
def load_predictor():
|
| 25 |
"""Load model predictor using torch.hub"""
|
| 26 |
-
predictor = torch.hub.load("hugoycj/StableNormal", "StableNormal", trust_repo=True
|
| 27 |
-
local_cache_dir='./weights')
|
| 28 |
return predictor
|
| 29 |
|
| 30 |
def process_image(
|
|
|
|
| 23 |
|
| 24 |
def load_predictor():
|
| 25 |
"""Load model predictor using torch.hub"""
|
| 26 |
+
predictor = torch.hub.load("hugoycj/StableNormal", "StableNormal", trust_repo=True)
|
|
|
|
| 27 |
return predictor
|
| 28 |
|
| 29 |
def process_image(
|