geninhu commited on
Commit
3f17a70
1 Parent(s): 2323db2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -34,7 +34,7 @@ def _denormalize(input: torch.Tensor) -> torch.Tensor:
34
  return (input * 127.5) + 127.5
35
 
36
  # Load generator
37
- generator = load_generator("fastgan-few-shot-shells")
38
 
39
  # Generate a random noise image
40
  noise = torch.zeros(1, 256, 1, 1, device=device).normal_(0.0, 1.0)
 
34
  return (input * 127.5) + 127.5
35
 
36
  # Load generator
37
+ generator = load_generator("huggan/fastgan-few-shot-shells")
38
 
39
  # Generate a random noise image
40
  noise = torch.zeros(1, 256, 1, 1, device=device).normal_(0.0, 1.0)