Update app.py

app.py

@@ -1,12 +1,22 @@
 import gradio as gr
-import json, os, vertexai
+import json, os, vertexai, wandb
 
 from dotenv import load_dotenv, find_dotenv
 _ = load_dotenv(find_dotenv())
 
+config = {
+    "model": "text-bison@001",
+    "temperature": 0,
+}
+
+wandb_api_key = os.environ["WANDB_API_KEY"]
 credentials = os.environ["CREDENTIALS"]
 project = os.environ["PROJECT"]
 
+wandb.login(key = wandb_api_key)
+wandb.init(project = "genai-txt", config = config)
+config = wandb.config
+
 credentials = json.loads(credentials)
 
 from google.oauth2 import service_account