bstraehle commited on
Commit
345228e
·
1 Parent(s): 2d5ceb5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -130,8 +130,9 @@ def invoke(openai_api_key, rag_option, prompt):
130
  if (prompt == ""):
131
  raise gr.Error("Prompt is required.")
132
  completion = ""
133
- #wandb.login(key = wandb_api_key)
134
- autolog({"project": "openai-llm-rag", "job_type": "generation"})
 
135
  try:
136
  llm = ChatOpenAI(model_name = config["model"],
137
  openai_api_key = openai_api_key,
 
130
  if (prompt == ""):
131
  raise gr.Error("Prompt is required.")
132
  completion = ""
133
+ wandb.login(key = wandb_api_key)
134
+ wandb.init(project = "openai-llm-rag", config = config)
135
+ autolog({"project": "openai-llm-rag"})
136
  try:
137
  llm = ChatOpenAI(model_name = config["model"],
138
  openai_api_key = openai_api_key,