Enable GPU

app.py

@@ -113,7 +113,7 @@ def check_nsfw_images(
     return images, has_nsfw_concepts
 
 # Function 
-#@spaces.GPU(enable_queue=True)
+@spaces.GPU(enable_queue=True)
 def generate_image(prompt, ckpt, aspect_ratio, mode):
     width, height = calculate_resolution(aspect_ratio, mode)  # Calculate resolution based on the aspect ratio
     checkpoint = checkpoints[ckpt][0]
@@ -172,4 +172,4 @@ with gr.Blocks(css="style.css") as demo:
     example_dropdown = gr.Dropdown(label='Select an Example', choices=[e["prompt"] for e in examples])
     example_dropdown.change(fn=set_example, inputs=example_dropdown, outputs=[prompt, ckpt, aspect, mode])
 
-demo.queue().launch()
+demo.queue().launch()