aicloud
/
hf_fully_body_anime_gan
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

modified: app.py
Build-Deploy-Actions Details

This commit is contained in:
jianjiang 2023-04-26 16:44:11 +08:00
parent 1fc7e9b8c0
commit 11315e551a
1 changed files with 13 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
app.py
View File

@ -1,3 +1,4 @@
#!coding:utf-8
import random
import gradio as gr
import imageio
@ -282,13 +283,13 @@ if __name__ == '__main__':
app = gr.Blocks()
with app:
gr.Markdown("# full-body anime GAN\n\n"
gr.Markdown("# AI人物动画照和视频生成\n\n"
"![visitor badge](https://visitor-badge.glitch.me/badge?page_id=skytnt.full-body-anime-gan)\n\n")
with gr.Tabs():
with gr.TabItem("generate image"):
with gr.TabItem("生成图片"):
with gr.Row():
with gr.Column():
gr.Markdown("generate image")
gr.Markdown("生成图片")
with gr.Row():
gen_input1 = gr.Slider(minimum=0, maximum=2 ** 32 - 1, step=1, value=0, label="seed")
gen_input2 = gr.Checkbox(label="Random", value=True)
@ -296,9 +297,9 @@ if __name__ == '__main__':
gen_input4 = gr.Slider(minimum=0, maximum=1, step=0.01, value=1, label="truncation psi 2")
gen_input5 = gr.Slider(minimum=0, maximum=1, step=0.01, value=1, label="noise strength")
with gr.Group():
gen_submit = gr.Button("Generate", variant="primary")
gen_submit = gr.Button("生成", variant="primary")
with gr.Column():
gen_output1 = gr.Image(label="output image")
gen_output1 = gr.Image(label="输出图像")
select_img_input_w1 = gr.Variable()
select_img_input_img1 = gr.Variable()
@ -310,7 +311,7 @@ if __name__ == '__main__':
examples_data = [[f"examples/{x:02d}.jpg"] for x in range(1, 5)]
encode_img_examples = gr.Dataset(components=[encode_img_input], samples=examples_data)
with gr.Group():
encode_img_submit = gr.Button("Run", variant="primary")
encode_img_submit = gr.Button("运行", variant="primary")
with gr.Column():
encode_img_output1 = gr.Textbox(label="output message")
with gr.Row():
@ -319,7 +320,7 @@ if __name__ == '__main__':
select_img_input_w2 = gr.Variable()
select_img_input_img2 = gr.Variable()
with gr.TabItem("generate video"):
with gr.TabItem("视频生成"):
with gr.Row():
with gr.Column():
gr.Markdown("generate video between 2 images")
@ -329,7 +330,7 @@ if __name__ == '__main__':
choices=["current generated image",
"current encoded image"], type="index")
with gr.Group():
select_img1_button = gr.Button("Select", variant="primary")
select_img1_button = gr.Button("选择", variant="primary")
select_img1_output_img = gr.Image(label="selected image 1")
select_img1_output_w = gr.Variable()
with gr.Column():
@ -337,14 +338,14 @@ if __name__ == '__main__':
choices=["current generated image",
"current encoded image"], type="index")
with gr.Group():
select_img2_button = gr.Button("Select", variant="primary")
select_img2_button = gr.Button("选择", variant="primary")
select_img2_output_img = gr.Image(label="selected image 2")
select_img2_output_w = gr.Variable()
generate_video_frame = gr.Slider(minimum=10, maximum=30, step=1, label="frame", value=15)
with gr.Group():
generate_video_button = gr.Button("Generate", variant="primary")
generate_video_button = gr.Button("生成", variant="primary")
with gr.Column():
generate_video_output = gr.Video(label="output video")
generate_video_output = gr.Video(label="输出视频")
gen_submit.click(gen_fn, [gen_input1, gen_input2, gen_input3, gen_input4, gen_input5],
[gen_output1, gen_input1, select_img_input_w1, select_img_input_img1])
encode_img_submit.click(encode_img_fn, [encode_img_input, gen_input5],
@ -362,4 +363,4 @@ if __name__ == '__main__':
generate_video_button.click(gen_video_fn,
[select_img1_output_w, select_img2_output_w, gen_input5, generate_video_frame],
[generate_video_output])
app.launch(server_name="0.0.0.0")
app.launch(server_name="0.0.0.0")