Update app.py

app.py

@@ -2,17 +2,19 @@ import zipfile
 from io import BytesIO
 
 import gradio as gr
-from diffusers import StableDiffusionPipeline
-from PIL import Image, ImageDraw, ImageFont
+from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
 import torch
+from PIL import Image, ImageDraw, ImageFont
 
 # =====================================================
-# Load Stable Diffusion Pipeline
+# Load Stable Diffusion with Euler Scheduler
 # =====================================================
+model_id = "stabilityai/stable-diffusion-2-1-base"
+
+scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
 pipe = StableDiffusionPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-2-1-base",
-    torch_dtype=torch.float16  # FP16 for GPU
-).to("cuda")  # change to "cpu" if GPU not available
+    model_id, scheduler=scheduler, torch_dtype=torch.float16
+).to("cuda")  # or "cpu" if GPU not available
 
 # =====================================================
 # Comic Styles & Descriptions