Spaces:
Running
Running
update
Browse files
app.py
CHANGED
|
@@ -248,8 +248,8 @@ def get_thumbnail(img):
|
|
| 248 |
return img_new
|
| 249 |
|
| 250 |
|
| 251 |
-
def gen_fn(
|
| 252 |
-
if
|
| 253 |
seed = random.randint(0, 2 ** 32 - 1)
|
| 254 |
z = RandomState(int(seed)).randn(1, 1024)
|
| 255 |
w = model.get_w(z.astype(dtype=np.float32), psi1, psi2)
|
|
@@ -289,9 +289,8 @@ if __name__ == '__main__':
|
|
| 289 |
with gr.Column():
|
| 290 |
gr.Markdown("generate image randomly or by seed")
|
| 291 |
with gr.Row():
|
| 292 |
-
gen_input1 = gr.
|
| 293 |
-
|
| 294 |
-
gen_input2 = gr.Slider(minimum=0, maximum=2 ** 32 - 1, step=1, value=0, label="seed")
|
| 295 |
gen_input3 = gr.Slider(minimum=0, maximum=1, step=0.01, value=1, label="truncation psi 1")
|
| 296 |
gen_input4 = gr.Slider(minimum=0, maximum=1, step=0.01, value=1, label="truncation psi 2")
|
| 297 |
gen_input5 = gr.Slider(minimum=0, maximum=1, step=0.01, value=1, label="noise strength")
|
|
@@ -346,7 +345,7 @@ if __name__ == '__main__':
|
|
| 346 |
with gr.Column():
|
| 347 |
generate_video_output = gr.Video(label="output video")
|
| 348 |
gen_submit.click(gen_fn, [gen_input1, gen_input2, gen_input3, gen_input4, gen_input5],
|
| 349 |
-
[gen_output1,
|
| 350 |
encode_img_submit.click(encode_img_fn, [encode_img_input, gen_input5],
|
| 351 |
[encode_img_output1, encode_img_output2, encode_img_output3, select_img_input_w2,
|
| 352 |
select_img_input_img2])
|
|
|
|
| 248 |
return img_new
|
| 249 |
|
| 250 |
|
| 251 |
+
def gen_fn(seed, random_seed, psi1, psi2, noise):
|
| 252 |
+
if random_seed:
|
| 253 |
seed = random.randint(0, 2 ** 32 - 1)
|
| 254 |
z = RandomState(int(seed)).randn(1, 1024)
|
| 255 |
w = model.get_w(z.astype(dtype=np.float32), psi1, psi2)
|
|
|
|
| 289 |
with gr.Column():
|
| 290 |
gr.Markdown("generate image randomly or by seed")
|
| 291 |
with gr.Row():
|
| 292 |
+
gen_input1 = gr.Slider(minimum=0, maximum=2 ** 32 - 1, step=1, value=0, label="seed")
|
| 293 |
+
gen_input2 = gr.Checkbox(label="Random", value=True)
|
|
|
|
| 294 |
gen_input3 = gr.Slider(minimum=0, maximum=1, step=0.01, value=1, label="truncation psi 1")
|
| 295 |
gen_input4 = gr.Slider(minimum=0, maximum=1, step=0.01, value=1, label="truncation psi 2")
|
| 296 |
gen_input5 = gr.Slider(minimum=0, maximum=1, step=0.01, value=1, label="noise strength")
|
|
|
|
| 345 |
with gr.Column():
|
| 346 |
generate_video_output = gr.Video(label="output video")
|
| 347 |
gen_submit.click(gen_fn, [gen_input1, gen_input2, gen_input3, gen_input4, gen_input5],
|
| 348 |
+
[gen_output1, gen_input1, select_img_input_w1, select_img_input_img1])
|
| 349 |
encode_img_submit.click(encode_img_fn, [encode_img_input, gen_input5],
|
| 350 |
[encode_img_output1, encode_img_output2, encode_img_output3, select_img_input_w2,
|
| 351 |
select_img_input_img2])
|