Update README.md

README.md

@@ -34,7 +34,7 @@ def _denormalize(input: torch.Tensor) -> torch.Tensor:
     return (input * 127.5) + 127.5
     
 # Load generator
-generator = load_generator("fastgan-few-shot-shells")
+generator = load_generator("huggan/fastgan-few-shot-shells")
 
 # Generate a random noise image
 noise = torch.zeros(1, 256, 1, 1, device=device).normal_(0.0, 1.0)