use gemini-2.0-flash model (#2648)

--------- Co-authored-by: rachelsaunders <52258509+rachelsaunders@users.noreply.github.com>

Author: thatfiredev

vertexai/app/src/main/kotlin/com/google/firebase/quickstart/vertexai/GenerativeAiViewModelFactory.kt

@@ -43,29 +43,29 @@ val GenerativeViewModelFactory = object : ViewModelProvider.Factory {
  return with(viewModelClass) {
  when {
  isAssignableFrom(SummarizeViewModel::class.java) -> {
- // Initialize a GenerativeModel with the `gemini-flash` AI model
+ // Initialize a GenerativeModel with a Gemini model
  // for text generation
  val generativeModel = Firebase.vertexAI.generativeModel(
- modelName = "gemini-1.5-flash",
+ modelName = "gemini-2.0-flash",
  generationConfig = config
  )
  SummarizeViewModel(generativeModel)
  }
 
  isAssignableFrom(PhotoReasoningViewModel::class.java) -> {
- // Initialize a GenerativeModel with the `gemini-flash` AI model
+ // Initialize a GenerativeModel with a Gemini model
  // for multimodal text generation
  val generativeModel = Firebase.vertexAI.generativeModel(
- modelName = "gemini-1.5-flash",
+ modelName = "gemini-2.0-flash",
  generationConfig = config
  )
  PhotoReasoningViewModel(generativeModel)
  }
 
  isAssignableFrom(ChatViewModel::class.java) -> {
- // Initialize a GenerativeModel with the `gemini-flash` AI model for chat
+ // Initialize a GenerativeModel with a Gemini model for chat
  val generativeModel = Firebase.vertexAI.generativeModel(
- modelName = "gemini-1.5-flash",
+ modelName = "gemini-2.0-flash",
  generationConfig = config
  )
  ChatViewModel(generativeModel)
@@ -88,19 +88,19 @@ val GenerativeViewModelFactory = object : ViewModelProvider.Factory {
  )
 
 
- // Initialize a GenerativeModel with the `gemini-pro` AI model for function calling chat
+ // Initialize a GenerativeModel with a Gemini model for function calling chat
  val generativeModel = Firebase.vertexAI.generativeModel(
- modelName = "gemini-1.5-flash",
+ modelName = "gemini-2.0-flash",
  generationConfig = config,
  tools = tools
  )
  FunctionsChatViewModel(generativeModel)
  }
 
  isAssignableFrom(AudioViewModel::class.java) -> {
- // Initialize a GenerativeModel with the `gemini-pro` AI model for audio generation
+ // Initialize a GenerativeModel with a Gemini model for audio understanding
  val generativeModel = Firebase.vertexAI.generativeModel(
- modelName = "gemini-1.5-pro-001",
+ modelName = "gemini-2.0-flash",
  generationConfig = config
  )
  AudioViewModel(generativeModel)