Update app.py

app.py

@@ -5,7 +5,6 @@ directory = pathlib.Path(os.getcwd())
 print(directory)
 sys.path.append(str(directory))
 import torch
-print(f"Is CUDA available: {torch.cuda.is_available()}")
 import numpy as np
 from omegaconf import OmegaConf
 from ldm.util import instantiate_from_config
@@ -36,8 +35,8 @@ def load_model_from_config(config, ckpt = None, verbose=True):
             print(u)
     else:
         print(f"Note chat no ckpt is loaded !!!")
-
-    model.cuda()
+    if torch.cuda.is_available():
+        model.cuda()
     model.eval()
     return model