fix: use cpu instead

app.py

@@ -6,7 +6,7 @@ from flask import (
     send_file,
 )
 from io import BytesIO
-from torch import float16
+from torch import cuda, float16
 from urllib.parse import unquote
 
 
@@ -26,7 +26,12 @@ def main():
         pretrained_model_or_path="Lykon/DreamShaper",
         torch_dtype=float16,
     )
-    pipeline.to("cuda")
+
+    if cuda.is_available():
+        pipeline.to("cuda")
+    
+    else:
+        pipeline.to("cpu")
     
     image = pipeline(
         unquote(string=prompt),