Update app.py

app.py

@@ -1,15 +1,11 @@
-import gradio as gr
-import torch
 from diffusers import StableDiffusionPipeline
-from PIL import Image
+import torch
 
-# Load model from Hugging Face hub (first time it downloads)
 pipe = StableDiffusionPipeline.from_pretrained(
-    "CompVis/stable-diffusion-v1-4",
-    torch_dtype=torch.float32,
-    revision="fp16",  # or "main"
-    use_auth_token=True
-).to("cpu")
+    "stabilityai/sdxl-turbo",
+    torch_dtype=torch.float16,  # or float32 if using CPU
+).to("cuda" if torch.cuda.is_available() else "cpu")
+
 
 
 def generate_image(prompt):