Update model/model_manager.py

model/model_manager.py

@@ -61,23 +61,23 @@ class ModelManager:
     
     @spaces.GPU(duration=120)
     def generate_image_ig(self, prompt, model_name):
-        if 'unsafe' not in self.NSFW_filter(prompt):
-            print('The prompt is safe')
-            pipe = self.load_model_pipe(model_name)
-            result = pipe(prompt=prompt)
-        else:
-            print(f'The prompt "{prompt}" is not safe')
-            result = ''
+        # if 'unsafe' not in self.NSFW_filter(prompt):
+        print('The prompt is safe')
+        pipe = self.load_model_pipe(model_name)
+        result = pipe(prompt=prompt)
+        # else:
+        #    print(f'The prompt "{prompt}" is not safe')
+        #    result = ''
         return result
 
     def generate_image_ig_api(self, prompt, model_name):
-        if 'unsafe' not in self.NSFW_filter(prompt):
-            print('The prompt is safe')
-            pipe = self.load_model_pipe(model_name)
-            result = pipe(prompt=prompt)
-        else:
-            print(f'The prompt "{prompt}" is not safe')
-            result = ''
+        # if 'unsafe' not in self.NSFW_filter(prompt):
+        print('The prompt is safe')
+        pipe = self.load_model_pipe(model_name)
+        result = pipe(prompt=prompt)
+        # else:
+        #    print(f'The prompt "{prompt}" is not safe')
+        #    result = ''
         return result
 
     def generate_image_ig_museum(self, model_name):