| import streamlit as st | |
| from PIL import Image | |
| import torch | |
| from diffusers import StableDiffusionPipeline | |
| import cv2 as cv | |
| st.title("Txt2Img") | |
| pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16") | |
| pipe.to("cpu") | |
| prom = st.text_area("Prompt",'') | |
| if len(prom)>5: | |
| image = pipe(prom).images[0] | |
| st.image(image) | |