mortal99/everything_can_coding

1# 1.安装依赖2import os3os.system("pip install paddlepaddle")4os.system("pip install paddlenlp==2.5.2")5os.system("pip install ppdiffusers==0.11.1")6 7# 2.导入库文件8import gradio as gr9from ppdiffusers import DiffusionPipeline, DPMSolverMultistepScheduler10import paddle11 12# 3.功能函数13def quickstart(prompt):14    image = pipe(prompt).images[0]15    return image16 17# 4.样式设计18with gr.Blocks() as demo:19    gr.Markdown("# 用 LoRA 和 DreamBooth 创作: 万物皆Coding")20    gr.Image("2100.png")21    gr.Markdown("## prompt: A picture of person coding")22    gr.Markdown("## prompt: A picture of dog coding")23    greet_btn = gr.Button("开始生成") 24    input_text=gr.Textbox(label="输入你想要的主体")   25    # 5.接口调用26    greet_btn.click(quickstart, inputs=input_text, outputs=gr.Image())27 28# 6.加载模型29pipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")30pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)31pipe.unet.load_attn_procs("mortal99", from_hf_hub=True)32 33# 7.启动34demo.launch()35