Spaces:
Sleeping
Sleeping
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(
|