Update app.py

app.py

@@ -2,14 +2,12 @@ import torch
 from diffusers import StableDiffusionPipeline
 import gradio as gr
 
-# Use GPU if available
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-# Load Stable Diffusion v1.5 from Hugging Face
+# Load Stable Diffusion v1.4 from Hugging Face
 pipe = StableDiffusionPipeline.from_pretrained(
-    "stable-diffusion-v1-5/stable-diffusion-v1-5",
+    "CompVis/stable-diffusion-v1-4",
     torch_dtype=torch.float16 if device == "cuda" else torch.float32,
-    revision="fp16" if device == "cuda" else None,
     use_safetensors=True
 )
 pipe = pipe.to(device)
@@ -20,10 +18,10 @@ def generate(prompt, guidance, steps, width, height):
     return image
 
 # Gradio UI
-title = "🎨 Offline Text-to-Image Generator (Stable Diffusion v1.5)"
+title = "🎨 Offline Text-to-Image Generator (Stable Diffusion v1.4)"
 description = "Generate images from text prompts using a fully self-hosted Stable Diffusion model."
 
-with gr.Blocks() as demo:
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="pink")) as demo:
     gr.Markdown(f"# {title}")
     gr.Markdown(description)