Update app.py

app.py

@@ -1,12 +1,11 @@
 import gradio as gr
-import boto3, json, os, wandb
+import boto3, json, os
 
 from dotenv import load_dotenv, find_dotenv
 _ = load_dotenv(find_dotenv())
 
 aws_access_key_id = os.environ["AWS_ACCESS_KEY_ID"]
 aws_secret_access_key = os.environ["AWS_SECRET_ACCESS_KEY"]
-wandb_api_key = os.environ["WANDB_API_KEY"]
 
 config = {
     "max_tokens_to_sample": 300,
@@ -23,12 +22,6 @@ bedrock_runtime = boto3.client(
     region_name = "us-west-2"
 )
 
-def wandb_log(prompt, completion):
-    wandb.login(key = wandb_api_key)
-    wandb.init(project = "bedrock-llm", config = config)
-    wandb.log({"prompt": str(prompt), "completion": str(completion)})
-    wandb.finish()
-
 def invoke(prompt):
     if (prompt == ""):
         raise gr.Error("Prompt is required.")
@@ -58,13 +51,11 @@ def invoke(prompt):
         completion = e
         
         raise gr.Error(e)
-    finally:
-        wandb_log(prompt, completion)
     
     return completion
 
 description = """<a href='https://www.gradio.app/'>Gradio</a> UI using the <a href='https://aws.amazon.com/bedrock/'>Amazon Bedrock</a> API 
-                 with <a href='https://www.anthropic.com/'>Anthropic</a> Claude 2 model. Evaluation via <a href='https://wandb.ai/bstraehle'>Weights & Biases</a>."""
+                 with <a href='https://www.anthropic.com/'>Anthropic</a> Claude 2 model."""
 
 gr.close_all()