Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -38,14 +38,14 @@ ARTICLE = '<center><img src="https://visitor-badge.glitch.me/badge?page_id=Time-
|
|
38 |
TOKEN = "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"
|
39 |
|
40 |
def image_create(seed: int, truncation_psi: float):
|
41 |
-
|
42 |
args=[str(input_path)],
|
43 |
namespace=Namespace(
|
44 |
encoder_ckpt=f"checkpoint/encoder/checkpoint_{spectral_sensitivity}.pt",
|
45 |
#gaussian=gaussian_radius,
|
46 |
log_visual_freq=1000
|
47 |
))
|
48 |
-
device = th.device()
|
49 |
generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args,device)
|
50 |
latent = torch.randn((1, 512), device=device)
|
51 |
img_out, _, _ = generator([latent])
|
|
|
38 |
TOKEN = "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"
|
39 |
|
40 |
def image_create(seed: int, truncation_psi: float):
|
41 |
+
args = ProjectorArguments().parse(
|
42 |
args=[str(input_path)],
|
43 |
namespace=Namespace(
|
44 |
encoder_ckpt=f"checkpoint/encoder/checkpoint_{spectral_sensitivity}.pt",
|
45 |
#gaussian=gaussian_radius,
|
46 |
log_visual_freq=1000
|
47 |
))
|
48 |
+
"""device = th.device()
|
49 |
generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args,device)
|
50 |
latent = torch.randn((1, 512), device=device)
|
51 |
img_out, _, _ = generator([latent])
|