@@ -56,6 +56,14 @@ class Prunaai(SimpleService):
         )
         if input_message.file:
             callback_data.update({'images': [input_message.file.url]})
+        if version == 'flux-fast' and (s_m := input_message.info.get('speed_mode', None)):
+            speed_mode = {
+                'Легкий сок 🍊 (более стабильный результат)': 'Lightly Juiced 🍊 (more consistent)',
+                'Сок 🔥 (режим по умолчанию)': 'Juiced 🔥 (default)',
+                'Экстра-сок 🔥 (быстрее генерация)': 'Extra Juiced 🔥 (more speed)',
+                'Мгновение ока 👁️ (максимальная скорость)': 'Blink of an eye 👁️',
+            }
+            callback_data.update({'speed_mode': speed_mode[s_m]})
         start_time = time.time()
         try:
             images = replicate_run(f'prunaai/{version}', callback_data)