Jinl commited on
Commit
e118776
1 Parent(s): 2b37c45

add NSFW checker and GPU mode

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -39,7 +39,7 @@ css = """
39
 
40
 
41
  class GlobalText:
42
- @spaces.GPU
43
  def __init__(self):
44
 
45
  # config dirs
@@ -52,8 +52,7 @@ class GlobalText:
52
 
53
  # self.savedir_mask = os.path.join(self.savedir, "mask")
54
 
55
- self.stable_diffusion_list = ["SimianLuo/LCM_Dreamshaper_v7"
56
- ]
57
  self.personalized_model_list = []
58
  self.lora_model_list = []
59
 
@@ -86,6 +85,7 @@ class GlobalText:
86
  self.max_results_index = len(self.results_paths) // 12
87
  return self.results_paths[0:12]
88
 
 
89
  def load_base_pipeline(self, model_path):
90
 
91
  time_start = datetime.now()
 
39
 
40
 
41
  class GlobalText:
42
+
43
  def __init__(self):
44
 
45
  # config dirs
 
52
 
53
  # self.savedir_mask = os.path.join(self.savedir, "mask")
54
 
55
+ self.stable_diffusion_list = ["SimianLuo/LCM_Dreamshaper_v7"]
 
56
  self.personalized_model_list = []
57
  self.lora_model_list = []
58
 
 
85
  self.max_results_index = len(self.results_paths) // 12
86
  return self.results_paths[0:12]
87
 
88
+ @spaces.GPU
89
  def load_base_pipeline(self, model_path):
90
 
91
  time_start = datetime.now()