File size: 922 Bytes
e2482cb |
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 28 29 30 31 32 33 |
import streamlit as st
from diffusers import DiffusionPipeline
from PIL import Image
# Load the model
@st.cache_resource
def load_pipeline():
return DiffusionPipeline.from_pretrained("IamCreateAI/Ruyi-Mini-7B")
pipe = load_pipeline()
# Streamlit App
st.title("AI Image Generator")
st.subheader("Generate stunning AI-powered images")
# Input for text prompt
prompt = st.text_input(
"Enter your prompt:",
"Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
)
if st.button("Generate Image"):
st.write("Generating image, please wait...")
try:
# Generate the image
image = pipe(prompt).images[0]
# Display the image
st.image(image, caption="Generated Image", use_column_width=True)
except Exception as e:
st.error(f"An error occurred: {e}")
st.write("Enter a creative prompt above and click the button to generate an AI-powered image!")
|