geekyrakshit commited on
Commit
64300e0
1 Parent(s): 85db9f4

update: chat app

Browse files
Files changed (1) hide show
  1. application_pages/chat_app.py +5 -2
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", value=""
 
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(project_name=f"{st.session_state.weave_entity_name}/{st.session_state.weave_project_name}")
 
 
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