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(
|