@@ -5,8 +5,10 @@ from io import BytesIO
 
 import requests
 from django.core.files import File
+from replicate.exceptions import ModelError
 
 from messages.models import Message
+from ml_model.exceptions import RequestBlocked, GenerationException
 from ml_model.services.base import SimpleService
 from ml_model.tasks import replicate_run
 
@@ -51,7 +53,12 @@ class Dalle(SimpleService):
                 **input_message.info,
             }
         )
-        images = replicate_run(self._CALLBACK, callback_data)
+        try:
+            images = replicate_run(self._CALLBACK, callback_data)
+        except ModelError as exc:
+            if any(error in str(exc) for error in ('E005', 'E006', 'sexual', 'NSFW')):
+                raise RequestBlocked
+            raise GenerationException from exc
         process_time = timedelta(seconds=(time.time() - start_time))
         self.handle_invoice(input_message.content_object.model, input_message=input_message)
         msgs = self.save_results(input_message.content, images, process_time, save)
@@ -7,8 +7,10 @@ from io import BytesIO
 import filetype
 import requests
 from django.core.files import File
+from replicate.exceptions import ModelError
 
 from messages.models import Message
+from ml_model.exceptions import RequestBlocked, GenerationException
 from ml_model.services.base import SimpleService
 from ml_model.tasks import replicate_run
 
@@ -60,10 +62,15 @@ class Hailuo(SimpleService):
             input_message.file.close()
             callback_data.update({'first_frame_image': image})
         start_time = time.time()
-        video = replicate_run(
-            f'minimax/{version}',
-            callback_data
-        )
+        try:
+            video = replicate_run(
+                f'minimax/{version}',
+                callback_data
+            )
+        except ModelError as exc:
+            if any(error in str(exc) for error in ('E005', 'E006', 'sexual')):
+                raise RequestBlocked
+            raise GenerationException from exc
         process_time = timedelta(seconds=(time.time() - start_time))
         self.handle_invoice(input_message.content_object.model, version=version, resolution=resolution)
         msgs = self.save_results(input_message.content, process_time, video, save)