@@ -13,9 +13,9 @@ from ml_model.services.kandinsky import Kandinsky
 from ml_model.services.lightning import Lightning
 from ml_model.services.llama import Llama
 from ml_model.services.logoai import Logoai
+from ml_model.services.midjourney import Midjourney
 from ml_model.services.mistral import Mistral
 from ml_model.services.musicgen import Musicgen
-from ml_model.services.openjourney import Openjourney
 from ml_model.services.pulid import Pulid
 from ml_model.services.recraft import Recraft
 from ml_model.services.sdxlemoji import Sdxlemoji
@@ -12,9 +12,9 @@ from ml_model.services.base import SimpleService
 from ml_model.tasks import replicate_run
 
 
-class Openjourney(SimpleService):
+class Midjourney(SimpleService):
     """
-    Openjourney Service
+    Midjourney Service
     contains abstract method make, which makes a generation
     """
 
@@ -29,16 +29,7 @@ class Openjourney(SimpleService):
             key='aspect_ratio',
             type=ModelParameter.TypeChoices.LIST,
             values={
-                'availables': [
-                    "1:1",
-                    "16:9",
-                    "4:3",
-                    "3:2",
-                    "2:3",
-                    "3:4",
-                    "9:16",
-                    "21:9"
-                ],
+                'availables': ['1:1', '16:9', '4:3', '3:2', '2:3', '3:4', '9:16', '21:9'],
                 'default': '1:1',
             },
         ),
@@ -89,6 +80,8 @@ class Openjourney(SimpleService):
         callback_data = dict(prompt=activation_prompt, **input_message.info)
         results = replicate_run(self._CALLBACK, callback_data)
         process_time = timedelta(seconds=(time.time() - start_time))
-        self.handle_invoice(input_message.content_object.model, input_message=input_message)
+        self.handle_invoice(
+            input_message.content_object.model, input_message=input_message
+        )
         msgs = self.save_results(input_message.content, results, process_time, save)
         return msgs