Spaces:
Runtime error
Runtime error
File size: 355 Bytes
67db81c 3712481 67db81c 7671d2b a543549 67db81c 4430ce1 0b87cae 7671d2b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
import torch
import streamlit as st
from diffusers import StableDiffusionPipeline
from PIL import Image
pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
pipe = pipe.to("cuda")
prompt = st.text_area("Enter a prompt: ")
image = pipe(prompt).images[0]
image = Image.fromarray(image)
image.show()
|