getImageModels method

Future<List<AIModel>> getImageModels()

Get models suitable for image generation

Implementation

Future<List<AIModel>> getImageModels() async {
  final allModels = await models();
  return allModels
      .where((model) =>
          model.id.contains('dall-e') || model.id.contains('dalle'))
      .toList();
}