Add files

app.py

@@ -252,6 +252,7 @@ def t2i_core(model, xt, emb, nemb, step=30, cfg_scale=7.5, return_list=False):
 ########
 
 class wrapper(object):
+    @spaces.GPU()
     def __init__(self, 
                  fp16=False, 
                  tag_diffuser=None, 
@@ -281,6 +282,7 @@ class wrapper(object):
         self.load_scheduler(tag_scheduler)
         return tag_diffuser, tag_lora, tag_scheduler
 
+    @spaces.GPU()
     def load_diffuser_lora(self, tag_diffuser, tag_lora):
         self.net = StableDiffusionPipeline.from_pretrained(
             choices.diffuser[tag_diffuser], torch_dtype=self.torch_dtype).to(self.device)