Update app.py

app.py

@@ -1,6 +1,3 @@
-import torch
-print(f"Is CUDA available: {torch.cuda.is_available()}")
-# print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
 import spaces
 import argparse, os, sys, glob
 import pathlib
@@ -8,6 +5,7 @@ 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