fixed to not display seeded fewshot as user message

app.py

@@ -360,15 +360,12 @@ if st.button("Generate Printer Issue Based on Seed Tasks"):
     model, tokenizer = get_mistral_model_tokenizer(sota_model_name)
     input_text = alpaca_input_text_format.format(customer_msg)
     st.write(f"```\n{input_text}```")
-    assistant_chat.write_stream(
-        stream_generation(
-            input_text,
-            show_prompt=False,
-            tokenizer=tokenizer,
-            model=model,
-            temperature=0.5,
-        )
-    )
+    few_shot_prompt_seeded = f"<s>[INST]{input_text}[/INST]\n"
+    st.markdown("Prompt:")
+    st.markdown(f"""```\n{few_shot_prompt_seeded}```""", unsafe_allow_html=True)
+
+    new_customer_msg = write_stream_user_chat_message(user_chat, model, tokenizer, few_shot_prompt_seeded)
+    st.session_state["user_msg_as_prompt"] = new_customer_msg
 
 # - Generate Customer Tweet
 if st.button("Generate Customer Message using Few Shots"):