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