Update README.md
Browse files
README.md
CHANGED
@@ -21,7 +21,7 @@ from PIL import Image
|
|
21 |
input_image = Image.open("path/to/your/image.jpg")
|
22 |
|
23 |
# Create predictor instance
|
24 |
-
predictor = torch.hub.load("
|
25 |
|
26 |
# Apply the model to the image
|
27 |
normal_image = predictor(input_image)
|
|
|
21 |
input_image = Image.open("path/to/your/image.jpg")
|
22 |
|
23 |
# Create predictor instance
|
24 |
+
predictor = torch.hub.load("hugoycj/StableNormal", "StableNormal_turbo", trust_repo=True, yoso_version='yoso-normal-v1-5')
|
25 |
|
26 |
# Apply the model to the image
|
27 |
normal_image = predictor(input_image)
|