Manage location on VertexAI instead of model (#12630)

This commit is contained in:
Paul Beusterien authored and Andrew Heard committed 2024-05-13 12:14:00 -04:00
1 parent fdc2985678
commit 0f1fdc888c
6 files changed
+33 -37

No files matched your search

@@ -32,10 +32,7 @@ class SummarizeViewModel: ObservableObject {
private var model: GenerativeModel?
init() {
model = VertexAI.vertexAI().generativeModel(
modelName: "gemini-1.0-pro",
location: "us-central1"
)
model = VertexAI.vertexAI(region: "us-central1").generativeModel(modelName: "gemini-1.0-pro")
}
func summarize(inputText: String) async {