Update model/model_manager.py

model/model_manager.py

@@ -44,19 +44,19 @@ class ModelManager:
     
     @spaces.GPU(duration=120)
     def generate_image_ig(self, prompt, model_name):
-        if self.NSFW_filter(prompt) == 'safe':
-            pipe = self.load_model_pipe(model_name)
-            result = pipe(prompt=prompt)
-        else:
-            result = ''
+        #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_api(self, prompt, model_name):
-        if self.NSFW_filter(prompt) == 'safe':
-            pipe = self.load_model_pipe(model_name)
-            result = pipe(prompt=prompt)
-        else:
-            result = ''
+        # 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):
@@ -119,11 +119,11 @@ class ModelManager:
 
     @spaces.GPU(duration=200)
     def generate_image_ie(self, textbox_source, textbox_target, textbox_instruct, source_image, model_name):
-        if self.NSFW_filter(" ".join([textbox_source, textbox_target, textbox_instruct])) == 'safe':
-            pipe = self.load_model_pipe(model_name)
-            result = pipe(src_image = source_image, src_prompt = textbox_source, target_prompt = textbox_target, instruct_prompt = textbox_instruct)
-        else:
-            result = ''
+        # if self.NSFW_filter(" ".join([textbox_source, textbox_target, textbox_instruct])) == 'safe':
+        pipe = self.load_model_pipe(model_name)
+        result = pipe(src_image = source_image, src_prompt = textbox_source, target_prompt = textbox_target, instruct_prompt = textbox_instruct)
+        # else:
+        #    result = ''
         return result
 
     def generate_image_ie_museum(self, model_name):