Update app.py

app.py

@@ -26,7 +26,7 @@ bedrock_runtime = boto3.client(
 def wandb_log(prompt, completion):
     wandb.login(key = wandb_api_key)
     wandb.init(project = "bedrock-llm", config = config)
-    wandb.log({"prompt": str(completion), "completion": str(completion)})
+    wandb.log({"prompt": str(prompt), "completion": str(completion)})
     wandb.finish()
 
 def invoke(prompt):