Update app.py

app.py

@@ -4,6 +4,18 @@ import torch
 
 from transformers import logging
 logging.set_verbosity_error()  # This suppresses warnings, including cache migration
+#GPU error
+from diffusers import StableDiffusionPipeline
+
+# Load the model
+model_id = "your-username/stable-diffusion-finetuned"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+
+# Check if CUDA is available and move the model to GPU if it is, otherwise use CPU
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe.to(device)
+
+print(f"Using device: {device}")
 
 
 # Load your fine-tuned model from Hugging Face