Update app.py
Browse files
app.py
CHANGED
@@ -132,8 +132,8 @@ def wandb_trace(prompt, completion, rag_option):
|
|
132 |
},
|
133 |
#start_time_ms=start_time_ms,
|
134 |
#end_time_ms=end_time_ms,
|
135 |
-
inputs={"prompt": prompt, "template":
|
136 |
-
outputs={"completion": completion},
|
137 |
)
|
138 |
trace.log("test")
|
139 |
wandb.finish()
|
|
|
132 |
},
|
133 |
#start_time_ms=start_time_ms,
|
134 |
#end_time_ms=end_time_ms,
|
135 |
+
inputs={"prompt": prompt, "template": "todo"},
|
136 |
+
outputs={"completion": json.loads(completion)},
|
137 |
)
|
138 |
trace.log("test")
|
139 |
wandb.finish()
|