Text2img / app.py
3laa2's picture
Update app.py
182d8d7
raw
history blame
429 Bytes
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')