Update app.py

app.py

@@ -13,7 +13,7 @@ config = {
 }
 
 wandb.login(key = wandb_api_key)
-wandb.init(project = "vertex-ai-txt", config = config)
+wandb.init(project = "vertex-ai-llm", config = config)
 config = wandb.config
 
 credentials = json.loads(credentials)
@@ -29,7 +29,7 @@ vertexai.init(project = project,
               credentials = credentials)
 
 from vertexai.language_models import TextGenerationModel
-generation_model = TextGenerationModel.from_pretrained("text-bison@001")
+generation_model = TextGenerationModel.from_pretrained(config.model)
 
 def invoke(prompt):
     completion = generation_model.predict(prompt = prompt).text