import streamlit as st
from diffusers import DiffusionPipeline

prompt = st.text_input('Describe the type of pokemon you would like to see:')
if(prompt):
    st.spinner(text='Generating your pokemon...')
    pipe = DiffusionPipeline.from_pretrained("justinpinkney/pokemon-stable-diffusion")
    image = pipe(prompt).images[0]
    if(image): 
        st.image(image)