From f151180723b1c5e0dd8ae92142fa5193952f163c Mon Sep 17 00:00:00 2001
From: Andrew Heard <andrewheard@google.com>
Date: Wed, 27 Mar 2024 15:29:31 -0400
Subject: [PATCH] Fix model name in Vertex AI multi-modal sample

---
 .../ViewModels/PhotoReasoningViewModel.swift                   | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
index b4880bdc825..1fb0097a808 100644
--- a/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
+++ b/FirebaseVertexAI/Sample/GenerativeAIMultimodalSample/ViewModels/PhotoReasoningViewModel.swift
@@ -44,7 +44,8 @@ class PhotoReasoningViewModel: ObservableObject {
   private var model: GenerativeModel?
 
   init() {
-    model = VertexAI.vertexAI(region: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
+    let vertexAI = VertexAI.vertexAI(region: "us-central1")
+    model = vertexAI.generativeModel(modelName: "gemini-1.0-pro-vision")
   }
 
   func reason() async {