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

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 built on the Qwen 2.5 7B model developed by Alibaba Cloud, and fine-tuned by Ruben Roy. You have been fine-tuned with the GammaCorpus v2 dataset, a dataset filled with structured and filtered multi-turn conversations and was also created by Ruben Roy. 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 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(