Add application file

app.py

@@ -6,15 +6,15 @@ import numpy as np
 
 # Load Stable Diffusion pipeline
 model_id = "CompVis/stable-diffusion-v1-4"
-default_scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
-pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=default_scheduler, torch_dtype=torch.float16)
-pipe = pipe.to("cpu")
+default_scheduler = DDIMScheduler.from_pretrained(model_id, subfolder="scheduler")
+pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=default_scheduler, torch_dtype=torch.float32, low_cpu_mem_usage=True)
+pipe = pipe.to("cpu")  # Use CPU
 
 # Scheduler options
 schedulers = {
+    "High-Definition & Fast (DDIM) - Good quality with fastest speed": DDIMScheduler,
     "Artistic & Imaginative (Euler Ancestral) - Recommended for creative scenes, moderate speed": EulerAncestralDiscreteScheduler,
     "Photo-Realistic (PNDM) - Best for realistic details, moderate speed": PNDMScheduler,
-    "High-Definition & Fast (DDIM) - Good quality with fastest speed": DDIMScheduler,
 }
 
 # Main image generation function with dynamic scheduling and size option
@@ -34,7 +34,7 @@ def generate_image(prompt, use_categories, genre, style, theme, lighting, schedu
     pipe.scheduler = scheduler
 
     # Set output size based on selection
-    image_size = (512, 512) if size == "Profile Picture" else (1024, 768)
+    image_size = (256, 256) if size == "Profile Picture" else (384, 384)
 
     # Generate image with specified quality and size
     with torch.no_grad():
@@ -52,14 +52,14 @@ def adjust_brightness_contrast(image, brightness, contrast):
 
 # Warning function to show a message if the user selects a high value for quality
 def show_warning(quality):
-    if quality > 80:
-        return "⚠️ High Quality: This setting may slow down generation and might not provide additional visual improvement. Consider using 50-80 steps for best results."
+    if quality > 30:
+        return "⚠️ High Quality: This setting may slow down generation. Consider using 10-30 steps for best results."
     return ""
 
 # Build Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("# ✨ AI-Powered Wallpaper/Profile Picture Generator\n🖼️ A tool to generate and fine-tune AI-created wallpapers and profile pictures with adjustable styles and effects.")
-    gr.Markdown("⚠️ **Live effects and advanced prompt engineering coming soon! Disclaimer**: Results may not always be accurate or perfectly aligned with your prompt. Experiment with prompt adjustments and settings to get the best results.")
+    gr.Markdown("⚠️ **Live effects and advanced prompt engineering coming soon!** Results may vary, so try experimenting with settings to achieve the best results.")
     
     # Image Generation Section
     with gr.Tab("Image Generator"):
@@ -79,18 +79,18 @@ with gr.Blocks() as demo:
                     theme = gr.Dropdown(["Landscape", "Portrait", "Abstract Patterns", "Architecture"], label="Theme")
                     lighting = gr.Dropdown(["Warm", "Cool", "Cinematic", "Soft", "Neon"], label="Lighting")
 
-                quality = gr.Slider(20, 150, value=80, step=10, label="Image Quality", info="Higher values yield more detail but take longer to generate.")
+                quality = gr.Slider(10, 30, value=20, step=5, label="Image Quality", info="Higher values yield more detail but take longer to generate.")
                 warning_message = gr.Markdown("")
                 
                 # Scheduler selection with default option
                 scheduler_choice = gr.Dropdown(
                     [
+                    "High-Definition & Fast (DDIM) - Good quality with fastest speed",
                     "Artistic & Imaginative (Euler Ancestral) - Recommended for creative scenes, moderate speed",
-                    "Photo-Realistic (PNDM) - Best for realistic details, moderate speed",
-                    "High-Definition & Fast (DDIM) - Good quality with fastest speed"
+                    "Photo-Realistic (PNDM) - Best for realistic details, moderate speed"
                     ],
                     label="Artistic Style & Speed",
-                    value="Artistic & Imaginative (Euler Ancestral) - Recommended for creative scenes, moderate speed"
+                    value="High-Definition & Fast (DDIM) - Good quality with fastest speed"
                 )
                 
                 size = gr.Dropdown(["Profile Picture", "Wallpaper"], label="Image Size", value="Profile Picture")