Update app.py

app.py

@@ -1,8 +1,13 @@
 import gradio as gr
 from diffusers import DDPMPipeline
+import torch 
 
 model = 'alexktrs/CumulusCloudsGenerator'
-device='cuda'
+
+if torch.cuda.is_available():
+    device='cuda'
+else:
+    device='cpu'
 
 generator = DDPMPipeline.from_pretrained(model)
 generator.to(device)