@@ -36,7 +36,8 @@ class NeuronModelSelector: 'image': 'images', 'audio': 'audio', 'video': 'videos', - 'code': 'code' + 'code': 'code', + 'voice': 'voice' } models = NeuronModel.objects.prefetch_related( Prefetch( @@ -5,13 +5,6 @@ from decimal import Decimal from io import BytesIO from typing import Any -from ml_model.exceptions import ( - RequestBlocked, - GenerationException, - ExceededContextLengthError, - ImageAnalysisError, -) - import filetype import requests from django.core.files import File @@ -23,8 +16,8 @@ from ml_model.tasks import replicate_run class Reve(SimpleService): PRICE = { - 'create': Decimal('7.5'), - 'edit-fast': Decimal('3') + 'create': Decimal('12.5'), + 'edit-fast': Decimal('5'), } @classmethod