Update app.py
Browse files
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
|
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(
|