Update app.py

app.py

@@ -128,6 +128,8 @@ def invoke(openai_api_key, rag_option, prompt):
     if (prompt == ""):
         raise gr.Error("Prompt is required.")
     completion = ""
+    wandb.login(key = wandb_api_key)
+    autolog({"project": "openai-llm-rag"})
     try:
         llm = ChatOpenAI(model_name = config["model"], 
                          openai_api_key = openai_api_key, 
@@ -150,7 +152,8 @@ def invoke(openai_api_key, rag_option, prompt):
         completion = e
         raise gr.Error(e)
     finally:
-        wandb_log(prompt, completion, rag_option)
+        #wandb_log(prompt, completion, rag_option)
+        wandb.finish()
     return completion
 
 description = """<strong>Overview:</strong> Context-aware multimodal reasoning application using a <strong>large language model (LLM)</strong> with