Fix: GPU usage on spaces

app.py

@@ -8,6 +8,7 @@ Date: Feb 26, 2025
 import os
 import torch
 import gradio as gr
+import spaces
 from tqdm.auto import tqdm
 import numpy as np
 from PIL import Image
@@ -16,10 +17,7 @@ from utils import (
     vignette_loss, get_concept_embedding, load_concept_library, image_grid
 )
 
-# Hugging Face Space configuration
-# Use @space decorator to configure the Space
-# This will set the Space to use zero GPU resources
-@gr.Blocks.add_decorator
+@spaces.GPU(enable_queue=True)
 def space(demo, **kwargs):
     demo.queue(concurrency_count=1, max_size=10)
     return demo
@@ -246,7 +244,7 @@ def generate_style_grid(prompt, seed=42, num_inference_steps=30, guidance_scale=
     return grid
 
 # Define Gradio interface
-@space
+@space.GPU(enable_queue=True)
 def create_demo():
     with gr.Blocks(title="Guided Stable Diffusion with Styles") as demo:
         gr.Markdown("# Guided Stable Diffusion with Styles")