Update sample app to use model 1.5-flash (#157)

google-gemini · May 28, 2024 · 8ba2c7d · 8ba2c7d
1 parent 77501f5
commit 8ba2c7d
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/...i-android-sample/app/src/main/kotlin/com/google/ai/sample/GenerativeAiViewModelFactory.kt b/...i-android-sample/app/src/main/kotlin/com/google/ai/sample/GenerativeAiViewModelFactory.kt
@@ -37,31 +37,31 @@ val GenerativeViewModelFactory = object : ViewModelProvider.Factory {
         return with(viewModelClass) {
             when {
                 isAssignableFrom(SummarizeViewModel::class.java) -> {
-                    // Initialize a GenerativeModel with the `gemini-pro` AI model
+                    // Initialize a GenerativeModel with the `gemini-flash` AI model
                     // for text generation
                     val generativeModel = GenerativeModel(
-                        modelName = "gemini-1.0-pro",
+                        modelName = "gemini-1.5-flash-latest",
                         apiKey = BuildConfig.apiKey,
                         generationConfig = config
                     )
                     SummarizeViewModel(generativeModel)
                 }
 
                 isAssignableFrom(PhotoReasoningViewModel::class.java) -> {
-                    // Initialize a GenerativeModel with the `gemini-pro-vision` AI model
+                    // Initialize a GenerativeModel with the `gemini-flash` AI model
                     // for multimodal text generation
                     val generativeModel = GenerativeModel(
-                        modelName = "gemini-1.0-pro-vision-latest",
+                        modelName = "gemini-1.5-flash-latest",
                         apiKey = BuildConfig.apiKey,
                         generationConfig = config
                     )
                     PhotoReasoningViewModel(generativeModel)
                 }
 
                 isAssignableFrom(ChatViewModel::class.java) -> {
-                    // Initialize a GenerativeModel with the `gemini-pro` AI model for chat
+                    // Initialize a GenerativeModel with the `gemini-flash` AI model for chat
                     val generativeModel = GenerativeModel(
-                        modelName = "gemini-1.0-pro",
+                        modelName = "gemini-1.5-flash-latest",
                         apiKey = BuildConfig.apiKey,
                         generationConfig = config
                     )