Add files

app.py

@@ -249,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, 
@@ -526,7 +526,8 @@ class wrapper(object):
             im_list += im if isinstance(im, list) else [im]
 
         return im_list
-
+    
+    @spaces.GPU()
     def run_iminvs(
             self, img, text, 
             cfg_scale, step, 
@@ -554,7 +555,8 @@ class wrapper(object):
         recon_output.save(ipath)
         
         return [recon_output]
-
+    
+    @spaces.GPU()
     def run_imedit(
             self, img, txt_0,txt_1, 
             threshold, cfg_scale, step, 
@@ -582,7 +584,7 @@ class wrapper(object):
      
         return [edited_img]
 
-
+    @spaces.GPU()
     def run_imintp(
             self, 
             img0, img1, txt0, txt1,