Update app.py

app.py

@@ -12,6 +12,12 @@ if huggingface_token is None:
 # Check if CUDA is available
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
+# Ensure GPU is available
+if device == "cuda":
+    print("CUDA is available. Using GPU.")
+else:
+    print("CUDA is not available. Using CPU.")
+
 model_path = snapshot_download(
     repo_id="stabilityai/stable-diffusion-3-medium", 
     revision="refs/pr/26",