Update app.py

app.py

@@ -20,6 +20,9 @@ import spaces  # Hugging Face Spaces module
 
 # Initialize the model
 pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-3.5-medium")
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe.to(device)
+model.to(device)