Spaces:
Sleeping
Sleeping
Commit
•
ca8d45b
1
Parent(s):
d375f86
fixed to use mistral
Browse files
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 =
|
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(
|