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