File size: 2,840 Bytes
74b9ce2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 |
import gradio as gr
from ppdiffusers import StableDiffusionPipeline
def generate_image(model, prompt, width, height, num_inference_steps, guidance_scale):
if model == "shanshui_gen_style":
pipe = StableDiffusionPipeline.from_pretrained("megemini/shanshui_gen_style", from_hf_hub=True)
elif model == "shanshui_style":
pipe = StableDiffusionPipeline.from_pretrained("megemini/shanshui_style", from_hf_hub=True)
else:
raise
image = pipe(
prompt,
num_inference_steps=100,
guidance_scale=7.5,
height=height,
width=width,).images[0]
return image
demo = gr.Blocks()
with demo:
gr.Markdown(
r"### 【Hackathon】基于PaddleNLP PPDiffusers 训练 AIGC 趣味模型"
)
gr.Markdown(
r"""
[【Hackathon】基于PaddleNLP PPDiffusers 训练 AIGC 趣味模型](https://github.com/PaddlePaddle/community/blob/master/hackthon_4th/%E3%80%90PaddlePaddle%20Hackathon%204%E3%80%91%20%E6%A8%A1%E5%9E%8B%E5%A5%97%E4%BB%B6%E5%BC%80%E6%BA%90%E8%B4%A1%E7%8C%AE%E4%BB%BB%E5%8A%A1%E5%90%88%E9%9B%86.md#no105%E5%9F%BA%E4%BA%8Epaddlenlp-ppdiffusers-%E8%AE%AD%E7%BB%83-aigc-%E8%B6%A3%E5%91%B3%E6%A8%A1%E5%9E%8B-)
"""
)
with gr.Row():
with gr.Column():
with gr.Row():
model = gr.Dropdown(["shanshui_gen_style", "shanshui_style"], label="Model", info="The model to generate image.")
with gr.Row():
prompt = gr.Textbox(label='prompt')
with gr.Row():
width = gr.Slider(128, 768, value=512, step=8, label="Width", info="The width of image.")
height = gr.Slider(128, 768, value=512, step=8, label="Height", info="The height of image.")
with gr.Row():
num_inference_steps = gr.Textbox(label='num inference steps')
guidance_scale = gr.Textbox(label='guidance scale')
with gr.Row():
btn = gr.Button(value="Run")
with gr.Column():
with gr.Row():
output = gr.Image()
gr.Examples(
[
[
"shanshui_gen_style",
"A fantasy landscape in <shanshui-gen-style>",
512,
288,
100,
7.5,
],
[
"shanshui_style",
"A fantasy landscape in <shanshui-style>",
512,
288,
100,
7.5,
],
],
[model, prompt, width, height, num_inference_steps, guidance_scale]
)
btn.click(
generate_image,
[model, prompt, width, height, num_inference_steps, guidance_scale],
output)
if __name__ == "__main__":
demo.launch(debug=True)
|