| import streamlit as st | |
| from PIL import Image | |
| import torch | |
| from diffusers import StableDiffusionPipeline | |
| import cv2 as cv | |
| st.title(" New App") | |
| 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] | |
| im = cv.imread(image) | |
| st.image(im) | |
| # else: | |
| # st.text_input("NNNN",'OOOOO') | |