Update model/model_manager.py

model/model_manager.py

@@ -49,6 +49,7 @@ class ModelManager:
     @spaces.GPU(duration=120)
     def generate_image_ig(self, prompt, model_name):
         if self.NSFW_filter(prompt) == 'safe':
+            print('The prompt is safe')
             pipe = self.load_model_pipe(model_name)
             result = pipe(prompt=prompt)
         else:
@@ -57,6 +58,7 @@ class ModelManager:
 
     def generate_image_ig_api(self, prompt, model_name):
         if self.NSFW_filter(prompt) == 'safe':
+            print('The prompt is safe')
             pipe = self.load_model_pipe(model_name)
             result = pipe(prompt=prompt)
         else: