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