Update app.py

app.py

@@ -12,6 +12,8 @@ import torch
 from diffusers import DiffusionPipeline
 
 DESCRIPTION = """# Playground v2.5"""
+if not torch.cuda.is_available():
+    DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"
 
 MAX_SEED = np.iinfo(np.int32).max
 CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "1") == "1"