Demo works with CPU

app.py

@@ -18,7 +18,7 @@ if (SPACE_ID := os.getenv('SPACE_ID')) is not None:
 if torch.cuda.is_available():
     DESCRIPTION += '\n<p>Running on GPU 🔥</p>'
 else:
-    DESCRIPTION += '\n<p>Running on CPU 🥶 This demo does not work on CPU.'
+    DESCRIPTION += '\n<p>Running on CPU 🥶'
 
 device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')