reduced the number of concepts and steps for inference

app.py

@@ -12,11 +12,11 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # List of concept embeddings compatible with SD v1.4
 concepts = [
-    "sd-concepts-library/cat-toy",
-    "sd-concepts-library/disco-diffusion-style",
-    "sd-concepts-library/modern-disney-style",
-    "sd-concepts-library/charliebo-artstyle",
-    "sd-concepts-library/redshift-render-style"
+    "sd-concepts-library/cat-toy"
+    # "sd-concepts-library/disco-diffusion-style",
+    # "sd-concepts-library/modern-disney-style",
+    # "sd-concepts-library/charliebo-artstyle",
+    # "sd-concepts-library/redshift-render-style"
 ]
 
 def download_concept_embedding(concept_name):
@@ -83,7 +83,7 @@ def generate_images(prompt):
             with autocast(device):
                 image = pipe(
                     concept_prompt,
-                    num_inference_steps=40,
+                    num_inference_steps=20,
                     generator=generator,
                     guidance_scale=7.5
                 ).images[0]