Robin Genolet commited on
Commit
07b6cfc
·
1 Parent(s): 7c3effb
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -225,6 +225,7 @@ def display_llm_output():
225
  st.session_state["prompt"] = prompt
226
  formatted_prompt = format_prompt(prompt_format, system_prompt, prompt)
227
  print(f"Formatted prompt: {format_prompt}")
 
228
  llm_response = get_llm_response(
229
  st.session_state["model_name_or_path"],
230
  st.session_state["temperature"],
 
225
  st.session_state["prompt"] = prompt
226
  formatted_prompt = format_prompt(prompt_format, system_prompt, prompt)
227
  print(f"Formatted prompt: {format_prompt}")
228
+ print(f"top_k: {st.session_state['top_k']}")
229
  llm_response = get_llm_response(
230
  st.session_state["model_name_or_path"],
231
  st.session_state["temperature"],