| 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") | |
| area = st.text_area("Area",'') | |
| if len(area)>5: | |
| image = pipe(area).images[0] | |
| st.image(image) | |