Update app.py

app.py

@@ -17,6 +17,7 @@ from langchain.vectorstores import MongoDBAtlasVectorSearch
 from pymongo import MongoClient
 
 from wandb.integration.openai import autolog
+autolog()
 
 from dotenv import load_dotenv, find_dotenv
 _ = load_dotenv(find_dotenv())
@@ -130,9 +131,9 @@ def invoke(openai_api_key, rag_option, prompt):
     if (prompt == ""):
         raise gr.Error("Prompt is required.")
     completion = ""
-    wandb.login(key = wandb_api_key)
-    wandb.init(project = "openai-llm-rag", config = config)
-    autolog({"project": "openai-llm-rag"})
+    #wandb.login(key = wandb_api_key)
+    #wandb.init(project = "openai-llm-rag", config = config)
+    #autolog({"project": "openai-llm-rag"})
     try:
         llm = ChatOpenAI(model_name = config["model"], 
                          openai_api_key = openai_api_key,