Update app.py
Browse files
app.py
CHANGED
@@ -135,7 +135,7 @@ def wandb_trace(rag_option, prompt, completion, status_msg, start_time_ms, end_t
|
|
135 |
"prompt_template": str((llm_template if (rag_option == "Off") else rag_template) if (str(status_msg) == "") else "")},
|
136 |
outputs = {"completion": str(completion)},
|
137 |
start_time_ms = start_time_ms,
|
138 |
-
end_time_ms = end_time_ms
|
139 |
)
|
140 |
trace.log("test")
|
141 |
wandb.finish()
|
|
|
135 |
"prompt_template": str((llm_template if (rag_option == "Off") else rag_template) if (str(status_msg) == "") else "")},
|
136 |
outputs = {"completion": str(completion)},
|
137 |
start_time_ms = start_time_ms,
|
138 |
+
end_time_ms = end_time_ms
|
139 |
)
|
140 |
trace.log("test")
|
141 |
wandb.finish()
|