Update README.md

README.md

@@ -55,7 +55,7 @@ Install with `pip install moderators` then run:
 ```python
 from moderators import AutoModerator
 
-model = AutoModerator.from_pretrained("viddexa/nsfw-nano")
+model = AutoModerator.from_pretrained("viddexa/nsfw-detection-nano")
 results = model("<path-to-image-file>")
 
 probs = {k: v for r in results for k, v in r.classifications.items()}
@@ -75,8 +75,8 @@ import torch
 
 img = Image.open("<path-to-image-file>")
 
-processor = AutoImageProcessor.from_pretrained("viddexa/nsfw-nano", use_fast = False)
-model=  AutoModelForImageClassification.from_pretrained("viddexa/nsfw-nano")
+processor = AutoImageProcessor.from_pretrained("viddexa/nsfw-detection-nano", use_fast = False)
+model=  AutoModelForImageClassification.from_pretrained("viddexa/nsfw-detection-nano")
 
 with torch.no_grad():
     inputs = processor(images=img, return_tensors="pt")