Update app.py

app.py

@@ -34,6 +34,10 @@ config = {
     "temperature": 0,
 }
 
+wandb.login(key = wandb_api_key)
+wandb.init(project = "openai-llm-rag", config = config)
+config = wand.config
+
 template = """If you don't know the answer, just say that you don't know, don't try to make up an answer. Keep the answer as concise as possible. Always say 
               "🧠 Thanks for using the app - Bernd" at the end of the answer. """
 
@@ -136,11 +140,6 @@ def invoke(openai_api_key, rag_option, prompt):
             completion = llm_chain(llm, prompt)
     except Exception as e:
         raise gr.Error(e)
-    wandb.login(key = wandb_api_key)
-    wandb.config[model] = model
-    wandb.config[temperature] = temperature
-    wandb.config[rag_option] = rag_option
-    wandb.init(project = "openai-llm-rag")
     wandb.log({"prompt": prompt, "completion": completion})
     return completion