| import os | |
| os.system("pip install paddlenlp==2.5.2") | |
| os.system("pip install ppdiffusers==0.11.1") | |
| from ppdiffusers import DiffusionPipeline, DPMSolverMultistepScheduler | |
| import paddle | |
| import streamlit as st | |
| st.header("用LoRA 和 DreamBooth画出你眼中的HomeTown") | |
| st.image("lora_outputs/validation_images/700.png") | |
| st.subheader("设置Prompt") | |
| st.write("输入你眼中家乡的几个关键句吧!看看能不能绘制出你心中的家乡!") | |
| pr1 = st.text_input('你心中的家乡：',"") | |
| test = st.button("⚡⚡开始生成⚡⚡") | |
| if test: | |
| pipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5") | |
| pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config) | |
| prompt = pr1 | |
| image = pipe(prompt).images[0] | |
| image.save("demo.png") | |
| st.image("demo.png") | |
| st.success('推理完毕！！', icon="✅") | |
| st.write("移动端长按图片可保存，PC端右键图片可另存到本地。") | |