Spaces:
Runtime error
Runtime error
Commit
·
64300e0
1
Parent(s):
85db9f4
update: chat app
Browse files
application_pages/chat_app.py
CHANGED
|
@@ -86,7 +86,8 @@ def initialize_guardrails():
|
|
| 86 |
)
|
| 87 |
elif guardrail_name == "PromptInjectionLlamaGuardrail":
|
| 88 |
llama_guard_checkpoint_name = st.sidebar.text_input(
|
| 89 |
-
"Checkpoint Name",
|
|
|
|
| 90 |
)
|
| 91 |
st.session_state.llama_guard_checkpoint_name = llama_guard_checkpoint_name
|
| 92 |
st.session_state.guardrails.append(
|
|
@@ -114,7 +115,9 @@ def initialize_guardrails():
|
|
| 114 |
|
| 115 |
|
| 116 |
if st.session_state.is_authenticated:
|
| 117 |
-
weave.init(
|
|
|
|
|
|
|
| 118 |
initialize_session_state()
|
| 119 |
st.title(":material/robot: Guardrails Genie Playground")
|
| 120 |
|
|
|
|
| 86 |
)
|
| 87 |
elif guardrail_name == "PromptInjectionLlamaGuardrail":
|
| 88 |
llama_guard_checkpoint_name = st.sidebar.text_input(
|
| 89 |
+
"Checkpoint Name",
|
| 90 |
+
value="wandb://geekyrakshit/guardrails-genie/ruk3f3b4-model:v8",
|
| 91 |
)
|
| 92 |
st.session_state.llama_guard_checkpoint_name = llama_guard_checkpoint_name
|
| 93 |
st.session_state.guardrails.append(
|
|
|
|
| 115 |
|
| 116 |
|
| 117 |
if st.session_state.is_authenticated:
|
| 118 |
+
weave.init(
|
| 119 |
+
project_name=f"{st.session_state.weave_entity_name}/{st.session_state.weave_project_name}"
|
| 120 |
+
)
|
| 121 |
initialize_session_state()
|
| 122 |
st.title(":material/robot: Guardrails Genie Playground")
|
| 123 |
|