Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -63,8 +63,7 @@ def image_create(input_img):
|
|
63 |
#gaussian=gaussian_radius,
|
64 |
log_visual_freq=1000,
|
65 |
input='text',
|
66 |
-
)
|
67 |
-
device = th.device()
|
68 |
generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args, device)
|
69 |
latent = torch.randn((1, 512), device=device)
|
70 |
img_out, _, _ = generator([latent])
|
@@ -79,8 +78,11 @@ def main():
|
|
79 |
# ini = "False1"
|
80 |
#result = subprocess.check_output(['nvidia-smi'])
|
81 |
args = parse_args()
|
|
|
|
|
|
|
82 |
iface = gr.Interface(
|
83 |
-
|
84 |
[
|
85 |
gr.inputs.Number(default=0, label='Seed'),
|
86 |
gr.inputs.Slider(
|
|
|
63 |
#gaussian=gaussian_radius,
|
64 |
log_visual_freq=1000,
|
65 |
input='text',
|
66 |
+
)
|
|
|
67 |
generator = create_generator("stylegan2-ffhq-config-f.pt","feng2022/Time-TravelRephotography_stylegan2-ffhq-config-f",args, device)
|
68 |
latent = torch.randn((1, 512), device=device)
|
69 |
img_out, _, _ = generator([latent])
|
|
|
78 |
# ini = "False1"
|
79 |
#result = subprocess.check_output(['nvidia-smi'])
|
80 |
args = parse_args()
|
81 |
+
device = th.device()
|
82 |
+
func = functools.partial(image_create, device=device)
|
83 |
+
func = functools.update_wrapper(func, generate_image)
|
84 |
iface = gr.Interface(
|
85 |
+
func,
|
86 |
[
|
87 |
gr.inputs.Number(default=0, label='Seed'),
|
88 |
gr.inputs.Slider(
|