Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -21,7 +21,8 @@ class Examples(gr.helpers.Examples):
|
|
21 |
|
22 |
def load_predictor():
|
23 |
"""Load model predictor using torch.hub"""
|
24 |
-
predictor = torch.hub.load("
|
|
|
25 |
return predictor
|
26 |
|
27 |
def process_image(
|
|
|
21 |
|
22 |
def load_predictor():
|
23 |
"""Load model predictor using torch.hub"""
|
24 |
+
predictor = torch.hub.load("Stable-X/StableNormal", "StableNormal_turbo",
|
25 |
+
trust_repo=True, yoso_version='yoso-normal-v1-8-1')
|
26 |
return predictor
|
27 |
|
28 |
def process_image(
|