jed-tiotuico commited on
Commit
ca8d45b
1 Parent(s): d375f86

fixed to use mistral

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -357,7 +357,7 @@ if st.button("Generate Printer Issue Based on Seed Tasks"):
357
  customer_msg = generate_printer_prompt(prompt_instructions)
358
  st.session_state["user_msg_as_prompt"] = customer_msg
359
  write_user_chat_message(user_chat, customer_msg)
360
- model, tokenizer = get_model_tokenizer(sota_model_name)
361
  input_text = alpaca_input_text_format.format(customer_msg)
362
  st.write(f"```\n{input_text}```")
363
  assistant_chat.write_stream(
 
357
  customer_msg = generate_printer_prompt(prompt_instructions)
358
  st.session_state["user_msg_as_prompt"] = customer_msg
359
  write_user_chat_message(user_chat, customer_msg)
360
+ model, tokenizer = get_mistral_model_tokenizer(sota_model_name)
361
  input_text = alpaca_input_text_format.format(customer_msg)
362
  st.write(f"```\n{input_text}```")
363
  assistant_chat.write_stream(