Spaces:
Sleeping
Sleeping
File size: 572 Bytes
52b8849 f62f83b 52b8849 15a1b0e 1161619 15a1b0e 86655c5 15a1b0e 86655c5 52b8849 174c59e 15a1b0e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
import streamlit as st
from diffusers import DDPMPipeline
import PIL
# Use the cache decorator to only re-run when the inputs change
# @st.cache(suppress_st_warning=True, allow_output_mutation=True)
def generate_image():
pipeline = DDPMPipeline.from_pretrained('dvoils/sd-class-butterflies-32')
image = pipeline().images[0]
return image
st.title('DDPMPipeline Image Generator')
if st.button('Generate Image'):
with st.spinner('Generating Image...'):
image = generate_image()
st.image(image, caption='Generated Image', use_column_width=True)
|