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.float16)
pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=True)


if len(area) > 5:
    image = pipe(area).images[0]
    im = Image.open(image)
    st.image(im)