Upload app.py

app.py

@@ -265,12 +265,12 @@ def generate_image(prompt, style_name, seed, num_inference_steps, guidance_scale
     except Exception as e:
         import traceback
         traceback.print_exc()
-        return None, f"Error: {str(e)}"
+        return [], f"Error: {str(e)}"
 
 def get_default_seed(style_name):
     if isinstance(style_name, str) and style_name in STYLES:
-        return gr.update(value=STYLES[style_name]["default_seed"])
-    return gr.update(value=42)
+        return STYLES[style_name]["default_seed"]
+    return 42
 
 def generate_all_styles(prompt, seed1, seed2, seed3, seed4, seed5, seed6, num_inference_steps, guidance_scale, contrast_scale, complexity_scale, vibrancy_scale, num_images_per_style):
     """Generate multiple images for all 6 styles with individual seeds"""
@@ -279,7 +279,7 @@ def generate_all_styles(prompt, seed1, seed2, seed3, seed4, seed5, seed6, num_in
     seeds = [seed1, seed2, seed3, seed4, seed5, seed6]
     for idx, (style_name, seed) in enumerate(zip(STYLES.keys(), seeds)):
         style_images, info = generate_image(prompt, style_name, seed, num_inference_steps, guidance_scale, contrast_scale=contrast_scale, complexity_scale=complexity_scale, vibrancy_scale=vibrancy_scale, num_images=num_images_per_style)
-        all_images.append(style_images)
+        all_images.append(style_images[:])
         info_texts.append(info)
     return all_images[0], all_images[1], all_images[2], all_images[3], all_images[4], all_images[5], "\n\n---\n\n".join(info_texts)
 
@@ -536,11 +536,17 @@ with gr.Blocks(title="Stable Diffusion Style Explorer", theme=gr.themes.Soft())
         gr.Markdown(f"**{style_name}**: {config['description']} | Token: `{config['token']}` | Default Seed: `{config['default_seed']}`")
 
 # Initialize pipeline on startup
-initialize_pipeline()
+#initialize_pipeline()
+
 
-# Enable queue globally for HF Spaces
-demo.queue()
 
 # Launch the app
 if __name__ == "__main__":
+    import gradio.routes
+
+    def disable_api_info(*args, **kwargs):
+        return {}
+
+    gradio.routes.get_api_info = disable_api_info
+    
     demo.launch(server_name="0.0.0.0", server_port=7860)