Spaces:

bstraehle
/

vertex-ai

Running

bstraehle commited on Dec 16, 2023

Commit

6cb526c

1 Parent(s): a57f48b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -33,7 +33,7 @@ vertexai.init(project = project,
 #from vertexai.language_models import TextGenerationModel
 #generation_model = TextGenerationModel.from_pretrained(config.model)
 from vertexai.preview.generative_models import GenerativeModel
-generation_model = GenerativeModel("gemini-pro")
 def invoke(prompt):
     #completion = generation_model.predict(prompt = prompt).text

 #from vertexai.language_models import TextGenerationModel
 #generation_model = TextGenerationModel.from_pretrained(config.model)
 from vertexai.preview.generative_models import GenerativeModel
+generation_model = GenerativeModel(config.model)
 def invoke(prompt):
     #completion = generation_model.predict(prompt = prompt).text