import gradio as gr
import spaces

@spaces.GPU
def generate_image(prompt, negative_prompt="", seed=42, width=512, height=512, guidance_scale=3.5, num_inference_steps=20):
    # This function will be called by gr.load(), so we don't need to implement it here
    pass

examples = [
    ["j0hnp0rqp1g-pigman as a camp counselor in the woods."],
    ["j0hnp0rqp1g-pigman dressed as batman"],
    ["j0hnp0rqp1g-pigman in a suit and tie"],
]

demo = gr.load("models/GenAIJake/j0hnp0rqp1g-pigman")

# Customize the interface
with demo:
    gr.Markdown("# j0hnp0rqp1g-pigman Image Generator")
    gr.Examples(examples, inputs=demo.input_components)

# Add GPUZero functionality
demo.queue()
demo.launch()