Spaces:
Sleeping
Sleeping
File size: 797 Bytes
45ee54e 1771de4 b1685eb bf1fe95 45ee54e b02f2f6 45ee54e bf1fe95 e4b219b 45ee54e e4b219b 45ee54e bf1fe95 4f953de 45ee54e bf1fe95 4f953de 45ee54e bf1fe95 45ee54e bf1fe95 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
import streamlit as st
from PIL import Image
import torch
from diffusers import StableDiffusionPipeline
def generate_image(input_text):
# Load Diffusion pipeline
pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float32)
# Generate image from text prompt
prompt = input_text
generated_image = pipe(prompt).images[0]
return generated_image
# Set Streamlit app title
st.title("Text to Image Generation App")
# Text input for prompt
input_text = st.text_input("Enter your image description:", "")
# Button to generate image
if st.button("Generate Image"):
# Generate image based on the prompt
img = generate_image(input_text)
# Display the generated image
st.image(img, caption="Generated Image")
|