add NSFW checker and GPU mode

app.py

@@ -17,6 +17,7 @@ from torchvision.utils import save_image
 import utils.ptp_utils as ptp_utils
 
 import torchvision.transforms as transforms
+import spaces
 
 try:
     import xformers
@@ -37,6 +38,7 @@ css = """
 
 
 class GlobalText:
+    @spaces.GPU
     def __init__(self):
         
         # config dirs
@@ -117,7 +119,7 @@ class GlobalText:
             print(f'load {base_model_dropdown} model success!')
             return gr.Dropdown()
 
-    
+    @spaces.GPU
     def generate(self, source, style, 
                        num_steps, co_feat_step,strength,
                        start_ac_layer, end_ac_layer,