add NSFW guard

model/model_manager.py

@@ -52,8 +52,11 @@ class ModelManager:
         return result
 
     def generate_image_ig_api(self, prompt, model_name):
-        pipe = self.load_model_pipe(model_name)
-        result = pipe(prompt=prompt)
+        if self.NSFW_filter(prompt) == 'safe':
+            pipe = self.load_model_pipe(model_name)
+            result = pipe(prompt=prompt)
+        else:
+            result = ''
         return result
 
     def generate_image_ig_museum(self, model_name):
@@ -192,8 +195,11 @@ class ModelManager:
         return result
 
     def generate_video_vg_api(self, prompt, model_name):
-        pipe = self.load_model_pipe(model_name)
-        result = pipe(prompt=prompt)
+        if self.NSFW_filter(prompt) == 'safe':
+            pipe = self.load_model_pipe(model_name)
+            result = pipe(prompt=prompt)
+        else:
+            result = ''
         return result
 
     def generate_video_vg_museum(self, model_name):