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 = [
    ["d3xt3r dachshund as a camp counselor in the woods."],
    ["d3xt3r dachshund dressed as batman"],
    ["d3xt3r dachshund in a suit and tie"],
]

demo = gr.load("models/GenAIJake/d3xt3r")

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

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