Text2img / app.py
3laa2's picture
Update app.py
46128eb
raw
history blame
428 Bytes
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(image)
# else:
# st.text_input("NNNN",'OOOOO')