rubenroy commited on
Commit
6ea832a
·
verified ·
1 Parent(s): 506450c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
14
  @spaces.GPU
15
  def generate(message, chat_history, temperature=0.7, top_p=0.9, top_k=50, max_new_tokens=512, repetition_penalty=1.1):
16
  messages = [
17
- {"role": "system", "content": "You are Zurich, a 7 billion parameter Large Language model. You have been trained with the GammaCorpus v2 dataset, a dataset filled with structured and filtered multi-turn conversations. You are a helpful assistant."},
18
  {"role": "user", "content": message}
19
  ]
20
  text = tokenizer.apply_chat_template(
 
14
  @spaces.GPU
15
  def generate(message, chat_history, temperature=0.7, top_p=0.9, top_k=50, max_new_tokens=512, repetition_penalty=1.1):
16
  messages = [
17
+ {"role": "system", "content": "You are a helpul assistant named Zurich, a 7 billion parameter Large Language model, you were fine-tuned and trained by Ruben Roy. You have been trained with the GammaCorpus v2 dataset, a dataset filled with structured and filtered multi-turn conversations, this was also made by Ruben Roy."}, # Attribution to Qwen is not included to prevent hallucinations.
18
  {"role": "user", "content": message}
19
  ]
20
  text = tokenizer.apply_chat_template(