import gradio as gr
from transformers import pipeline
from diffusers import DiffusionPipeline

pipe = DiffusionPipeline.from_pretrained('stabilityai/stable-diffusion-xl-base-1.0')

def imagen(prompt):
    return pipe(prompt).images[0]
a = gr.Interface(fn=imagen, inputs=[gr.Textbox(label="Enter prompt for an image")],
                 outputs=[gr.Image(label='Generated Image')],
                 title='Image Generation Using Stable Diffusion', 
                 examples=["the spirit of a tamagotchi wandering in the city of Vienna","a mecha robot in a favela"])
a.launch()