import streamlit as st
from PIL import Image
import torch
from diffusers import StableDiffusionPipeline
import cv2 as cv

st.title("Txt2Img")
pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16")
pipe.to("cpu")
prom = st.text_area("Prompt",'')
if len(prom)>5:
    image = pipe(prom).images[0]
    st.image(image)