Add files

app.py

@@ -26,6 +26,7 @@ from diffusers import DDIMScheduler
 from app_utils import auto_dropdown
 
 from huggingface_hub import hf_hub_download
+import spaces
 
 version = "Smooth Diffusion Demo v1.0"
 refresh_symbol = "\U0001f504" # 🔄
@@ -248,7 +249,7 @@ def t2i_core(model, xt, emb, nemb, step=30, cfg_scale=7.5, return_list=False):
 ########
 # main #
 ########
-
+@spaces.GPU()
 class wrapper(object):
     def __init__(self, 
                  fp16=False, 
@@ -256,8 +257,7 @@ class wrapper(object):
                  tag_lora=None,
                  tag_scheduler=None,):
 
-        # self.device = "cuda"
-        self.device = "cpu"
+        self.device = "cuda:0" if torch.cuda.is_available() else "cpu"
         if fp16:
             self.torch_dtype = torch.float16
         else: