Upload app.py

app.py

@@ -15,7 +15,8 @@ from utils import (
     load_models, clear_gpu_memory, set_timesteps, latents_to_pil, 
     vignette_loss, get_concept_embedding, image_grid
 )
-from diffusers import StableDiffusionPipeline
+# Remove this import to avoid the cached_download error
+# from diffusers import StableDiffusionPipeline
 
 
 def generate_latents(prompt, seed, num_inference_steps, guidance_scale, vignette_loss_scale, concept, concept_strength, height, width):
@@ -146,8 +147,8 @@ def generate_style_images(prompt, num_inference_steps=30, guidance_scale=7.5,
     concept_labels = []
 
     # Load and remove the "none" element
-    concepts_list = art_concepts.keys()
-    concepts_list.pop()
+    concepts_list = list(art_concepts.keys())
+    concepts_list.remove("none")
     
     for seed_no, concept in zip(seed_list, concepts_list):
         # Clear the CUDA cache
@@ -178,6 +179,8 @@ def create_demo():
             with gr.Row():
                 with gr.Column():
                     all_styles = ["none"] + list(art_concepts.keys())
+                    all_styles.remove("none")  # Remove "none" to avoid duplication
+                    all_styles = ["none"] + all_styles  # Add it back at the beginning
 
                     prompt = gr.Textbox(label="Prompt", placeholder="A cat sitting on a chair")
                     seed = gr.Slider(minimum=0, maximum=10000, step=1, label="Seed", value=1000)