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(
|