jed-tiotuico commited on
Commit
0983bd0
1 Parent(s): 4cb2b96

fixed to not display as user message pt 2

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -355,8 +355,7 @@ if st.button("Generate Printer Issue Based on Seed Tasks"):
355
  prompt_instructions += random.sample(seed_instructions, num_prompt_instructions - len(prompt_instructions))
356
  random.shuffle(prompt_instructions)
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}```")
 
355
  prompt_instructions += random.sample(seed_instructions, num_prompt_instructions - len(prompt_instructions))
356
  random.shuffle(prompt_instructions)
357
  customer_msg = generate_printer_prompt(prompt_instructions)
358
+ # instantiate the model and tokenizer
 
359
  model, tokenizer = get_mistral_model_tokenizer(sota_model_name)
360
  input_text = alpaca_input_text_format.format(customer_msg)
361
  st.write(f"```\n{input_text}```")