Spaces:

bstraehle
/

rag

Running

bstraehle commited on Dec 20, 2023

Commit

5e334ff

1 Parent(s): 9944aa6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -139,7 +139,7 @@ def wandb_trace(rag_option, prompt, completion, status_msg, start_time_ms, end_t
         },
         inputs = {["rag_option": rag_option if (str(status_msg) == "") else ""],
                   ["prompt": str(prompt if (str(status_msg) == "") else "")],
-                  ["prompt_template": str((llm_template if (rag_option == "Off") else rag_template) if (str(status_msg) == "") else "")}],
         outputs = {["result": result],
                    ["document_0": "" if (rag_option == "Off") else str(document_0)],
                    ["document_1": "" if (rag_option == "Off") else str(document_1)],

         },
         inputs = {["rag_option": rag_option if (str(status_msg) == "") else ""],
                   ["prompt": str(prompt if (str(status_msg) == "") else "")],
+                  ["prompt_template": str((llm_template if (rag_option == "Off") else rag_template) if (str(status_msg) == "") else "")]},
         outputs = {["result": result],
                    ["document_0": "" if (rag_option == "Off") else str(document_0)],
                    ["document_1": "" if (rag_option == "Off") else str(document_1)],