Check if cuda is available

app.py

@@ -7,7 +7,10 @@ pipe = StableDiffusionPipeline.from_pretrained(
     safety_checker=None,
     torch_dtype=torch.float16,
     use_safetensors=True,
-).to("cuda")
+)
+
+if torch.cuda.is_available():
+    pipe.to("cuda")
 
 
 def generate_image(text):