Stable-X commited on
Commit
535c9db
1 Parent(s): 5c8adfd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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("Stable-X/StableNormal", "StableNormal_turbo", trust_repo=True, yoso_version='yoso-normal-v1-4')
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)