getModelCapabilities method

Future<Map<String, bool>> getModelCapabilities(
  1. String modelId
)

Get model capabilities

Implementation

Future<Map<String, bool>> getModelCapabilities(String modelId) async {
  final modelInfo = await getModelInfo(modelId);
  if (modelInfo == null) {
    return {
      'tts': false,
      'stt': false,
      'voice_conversion': false,
      'voice_cloning': false,
    };
  }

  return {
    'tts': modelInfo['can_do_text_to_speech'] as bool? ?? false,
    'stt': modelInfo['can_do_voice_conversion'] as bool? ?? false,
    'voice_conversion':
        modelInfo['can_do_voice_conversion'] as bool? ?? false,
    'voice_cloning': modelInfo['can_be_finetuned'] as bool? ?? false,
  };
}