JamalAG commited on
Commit
9d05b30
·
1 Parent(s): 9402327

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ user_message = st.text_input("You:", "")
14
 
15
  if st.button("Send"):
16
  # Use TinyLlama chatbot pipeline to generate a response
17
- messages = [{"role": "system", "content": "You are a friendly chatbot who always responds in the style of a pirate"},
18
  {"role": "user", "content": user_message}]
19
  prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
20
  response = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)[0]["generated_text"]
 
14
 
15
  if st.button("Send"):
16
  # Use TinyLlama chatbot pipeline to generate a response
17
+ messages = [{"role": "system", "content": "You are a friendly chatbot who always responds in the style of a Doctor"},
18
  {"role": "user", "content": user_message}]
19
  prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
20
  response = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)[0]["generated_text"]