Skip to content

Commit b794bea

Browse files
I am now adding the new model choices to MenuScreen.
1 parent 45549ba commit b794bea

2 files changed

Lines changed: 6 additions & 4 deletions

File tree

app/src/main/kotlin/com/google/ai/sample/GenerativeAiViewModelFactory.kt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,10 @@ import com.google.ai.sample.feature.multimodal.PhotoReasoningViewModel
1313
enum class ModelOption(val displayName: String, val modelName: String) {
1414
GEMINI_FLASH_LITE("Gemini 2.0 Flash Lite", "gemini-2.0-flash-lite"),
1515
GEMINI_FLASH("Gemini 2.0 Flash", "gemini-2.0-flash"),
16-
GEMINI_FLASH_LITE_PREVIEW("Gemini 2.5 Flash Lite Preview", "gemini-2.5-flash-lite-preview-06-17"),
17-
GEMINI_FLASH_PREVIEW("Gemini 2.5 Flash", "gemini-2.5-flash"),
16+
GEMINI_FLASH_PREVIEW("Gemini 2.5 Flash Preview", "gemini-2.5-flash"),
1817
GEMINI_PRO("Gemini 2.5 Pro (maybe don't work)", "gemini-2.5-pro"),
19-
GEMMA_3N_E4B_IT("Gemma 3n E4B it", "gemma-3n-e4b-it")
18+
GEMINI_FLASH_LITE_PREVIEW("Gemini 2.5 Flash Lite Preview", "gemini-2.5-flash-lite-preview-06-17"),
19+
GEMMA_3N_E4B_IT("Gemma 3N E4B IT", "gemma-3n-e4b-it")
2020
}
2121

2222
val GenerativeViewModelFactory = object : ViewModelProvider.Factory {

app/src/main/kotlin/com/google/ai/sample/MenuScreen.kt

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -145,7 +145,9 @@ fun MenuScreen(
145145
ModelOption.GEMINI_FLASH_LITE,
146146
ModelOption.GEMINI_FLASH,
147147
ModelOption.GEMINI_FLASH_PREVIEW,
148-
ModelOption.GEMINI_PRO
148+
ModelOption.GEMINI_PRO,
149+
ModelOption.GEMINI_FLASH_LITE_PREVIEW,
150+
ModelOption.GEMMA_3N_E4B_IT
149151
)
150152

151153
orderedModels.forEach { modelOption ->

0 commit comments

Comments
 (0)