bstraehle commited on
Commit
a36a82c
·
1 Parent(s): a84efc1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ generation_model = GenerativeModel(config["model"])
38
  def wandb_log(prompt, completion):
39
  wandb.login(key = wandb_api_key)
40
  wandb.init(project = "vertex-ai-llm", config = config)
41
- wandb.log({"prompt": prompt, "completion": completion})
42
  wandb.finish()
43
 
44
  def invoke(prompt):
 
38
  def wandb_log(prompt, completion):
39
  wandb.login(key = wandb_api_key)
40
  wandb.init(project = "vertex-ai-llm", config = config)
41
+ wandb.log({"prompt": str(prompt), "completion": str(completion)})
42
  wandb.finish()
43
 
44
  def invoke(prompt):