Spaces:
Running
Running
geekyrakshit
commited on
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 |
|