Update app.py

app.py

@@ -4,14 +4,18 @@ import boto3, json, os, wandb
 from dotenv import load_dotenv, find_dotenv
 _ = load_dotenv(find_dotenv())
 
+wandb_api_key = os.environ["WANDB_API_KEY"]
+aws_access_key_id = os.environ["AWS_ACCESS_KEY_ID"]
+aws_secret_access_key = os.environ["AWS_SECRET_ACCESS_KEY"]
+
 config = {
     "model": "anthropic.claude-v2",
     "temperature": 1,
 }
 
-wandb_api_key = os.environ["WANDB_API_KEY"]
-aws_access_key_id = os.environ["AWS_ACCESS_KEY_ID"]
-aws_secret_access_key = os.environ["AWS_SECRET_ACCESS_KEY"]
+wandb.login(key = wandb_api_key)
+wandb.init(project = "genai-txt", config = config)
+config = wandb.config
 
 bedrock_runtime = boto3.client(
     aws_access_key_id = aws_access_key_id,