File size: 355 Bytes
64ea77f 2691f07 038d520 25ab162 64ea77f a0b3ec0 f557859 182d8d7 420d9c0 46128eb 2691f07 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
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")
area = st.text_area("Area",'')
if len(area)>5:
image = pipe(area).images[0]
st.image(image)
|