Update app.py

app.py

@@ -20,8 +20,8 @@ You can try out the prompts and check for yourself.
 
 You can check out more of my spaces. Demo by [Sunder Ali Khowaja](https://sander-ali.github.io) - [Github](https://github.com/sander-ali)
 """
-# if not torch.cuda.is_available():
-#     DESCRIPTION += "\n<h1>Running on CPU 🥶 This demo does not work on CPU. </h1>"
+if not torch.cuda.is_available():
+    MARKDOWN += "\n<h1>Running on CPU 🥶 This demo does not work on CPU. </h1>"
 
 MAX_SEED = np.iinfo(np.int32).max
 CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "0") == "1"