@@ -59,6 +59,7 @@ class Pruna_V(SimpleService): { 'prompt': self.translate_prompt(input_message.content), 'draft': generation_mode == 'draft', + 'disable_safety_filter': False, **input_message.info, } ) @@ -75,6 +76,8 @@ class Pruna_V(SimpleService): start_time = time.time() try: images = replicate_run('prunaai/p-video', callback_data) + if 'nsfw.jpeg' == str(images).split('/')[-1]: + raise RequestBlocked except ModelError as exc: if any(error in str(exc) for error in ('E005', 'E006', 'sexual', 'NSFW')): raise RequestBlocked