Update app.py

app.py

@@ -2,24 +2,20 @@ from diffusers import StableDiffusionPipeline
 import gradio as gr
 import torch
 import os
-
 from huggingface_hub import login
-
-# 3. Login to Hugging Face (replace YOUR_HF_TOKEN with your actual token)
-
+from spaces import GPU  # 👈 only for HF Spaces
 
 login(token=os.getenv("HF_TOKEN"))
 
-# 4. Load private model using your token
-model_id ="stabilityai/stable-diffusion-2-1-base"  
-
+model_id = "stabilityai/stable-diffusion-2-1-base"  
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 pipe = StableDiffusionPipeline.from_pretrained(
     model_id,
-    torch_dtype=torch.float16  # smaller memory
+    torch_dtype=torch.float16
 ).to(device)
 
+@GPU  # 👈 this tells HF Space to start with GPU
 def generate_image(prompt):
     image = pipe(prompt).images[0]
     return image
@@ -31,4 +27,4 @@ iface = gr.Interface(
     title="Private Stable Diffusion Demo"
 )
 
-iface.launch(share=True)
+iface.launch()