Update app.py

app.py

@@ -1,37 +1,40 @@
 import gradio as gr
-from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
 import torch
+from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
 
-# Load the model
-model_id = "runwayml/stable-diffusion-v1-5"
-device = "cuda" if torch.cuda.is_available() else "cpu"
+# ---- Configuration ---- #
+MODEL_ID = "runwayml/stable-diffusion-v1-5"
+USE_CUDA = torch.cuda.is_available()
+DEVICE = "cuda" if USE_CUDA else "cpu"
+DTYPE = torch.float16 if USE_CUDA else torch.float32
 
-# Load the pipeline
+# ---- Load Stable Diffusion Pipeline ---- #
 pipe = StableDiffusionPipeline.from_pretrained(
-    model_id,
-    torch_dtype=torch.float16 if device == "cuda" else torch.float32
-)
-
-# ✅ Move to device
-pipe = pipe.to(device)
+    MODEL_ID,
+    torch_dtype=DTYPE
+).to(DEVICE)
 
-# ✅ Change the scheduler *after* pipe is defined
+# ---- Swap the Scheduler for Better Quality ---- #
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 
-# Inference function
+# ---- Image Generation Function ---- #
 def generate_image(prompt):
-    enhanced_prompt = f"{prompt}, highly detailed, 8k, photorealistic, ultra-realistic lighting"
-    image = pipe(enhanced_prompt).images[0]
-    return image
+    enhanced_prompt = f"{prompt}, ultra realistic, high detail, 8k resolution, DSLR photography, natural lighting"
+    result = pipe(enhanced_prompt)
+    return result.images[0]
 
-# Gradio UI
+# ---- Gradio Interface ---- #
 demo = gr.Interface(
     fn=generate_image,
-    inputs=gr.Textbox(label="Enter image description", placeholder="e.g. A magical tree in a glowing forest"),
+    inputs=gr.Textbox(
+        label="Enter image description",
+        placeholder="e.g. A cozy cabin in a snowy forest"
+    ),
     outputs=gr.Image(type="pil"),
-    title="Text to Image Generator",
-    description="Enter a prompt and generate a photorealistic image using Stable Diffusion"
+    title="🖼️ Realistic Text-to-Image Generator",
+    description="Generate high-quality, photorealistic images using Stable Diffusion v1.5 + DPM Scheduler"
 )
 
+# ---- Launch App ---- #
 if __name__ == "__main__":
     demo.launch()