bstraehle commited on
Commit
328038c
1 Parent(s): 9e8e54f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ bedrock_runtime = boto3.client(
26
  def wandb_log(prompt, completion):
27
  wandb.login(key = wandb_api_key)
28
  wandb.init(project = "bedrock-llm", config = config)
29
- wandb.log({"prompt": prompt, "completion": completion})
30
  wandb.finish()
31
 
32
  def invoke(prompt):
 
26
  def wandb_log(prompt, completion):
27
  wandb.login(key = wandb_api_key)
28
  wandb.init(project = "bedrock-llm", config = config)
29
+ wandb.log({"prompt": str(completion), "completion": str(completion)})
30
  wandb.finish()
31
 
32
  def invoke(prompt):