Update app.py

app.py

@@ -1,9 +1,15 @@
 import gradio as gr
-import boto3, json, os
+import boto3, json, os, wandb
 
 from dotenv import load_dotenv, find_dotenv
 _ = load_dotenv(find_dotenv())
 
+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"]
 
@@ -18,7 +24,7 @@ bedrock_runtime = boto3.client(
 def invoke(prompt):
     body = json.dumps({"prompt": "\n\nHuman: " + prompt + "\n\nAssistant: ",
                        "max_tokens_to_sample": 300,
-                       "temperature": 1,
+                       "temperature": config.temperature,
                        "top_k": 250,
                        "top_p": 0.999,
                        "stop_sequences": ["\n\nHuman: "]