Upload app.py

app.py

@@ -17,10 +17,6 @@ from utils import (
     vignette_loss, get_concept_embedding, load_concept_library, image_grid
 )
 
-@spaces.GPU(enable_queue=True)
-def space(demo, **kwargs):
-    demo.queue(concurrency_count=1, max_size=10)
-    return demo
 
 # Set device
 device = "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"
@@ -244,7 +240,7 @@ def generate_style_grid(prompt, seed=42, num_inference_steps=30, guidance_scale=
     return grid
 
 # Define Gradio interface
-@space.GPU(enable_queue=True)
+@spaces.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")