| import streamlit as st | |
| from PIL import Image | |
| import torch | |
| from diffusers import StableDiffusionPipeline | |
| st.title(" New App") | |
| area = st.text_area("Area",'') | |
| pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float32) | |
| if len(area) > 5: | |
| image = pipe(area).images | |
| print(len(image),type(image),image.shape) | |
| im = Image.open(image[0]) | |
| st.image(im) | |