Spaces:
Sleeping
Sleeping
Commit
·
345f8c1
1
Parent(s):
251a8f8
chore: test duration
Browse files
app.py
CHANGED
|
@@ -14,7 +14,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
|
|
| 14 |
|
| 15 |
classifier = pipeline("text-classification", model="saiteki-kai/QA-DeBERTa-v3-large-threshold-v2")
|
| 16 |
|
| 17 |
-
@spaces.GPU(duration=
|
| 18 |
def generate(message):
|
| 19 |
messages = [
|
| 20 |
{"role": "system", "content": "You are a helpul assistant named Zurich, a 14 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.
|
|
|
|
| 14 |
|
| 15 |
classifier = pipeline("text-classification", model="saiteki-kai/QA-DeBERTa-v3-large-threshold-v2")
|
| 16 |
|
| 17 |
+
@spaces.GPU(duration=60)
|
| 18 |
def generate(message):
|
| 19 |
messages = [
|
| 20 |
{"role": "system", "content": "You are a helpul assistant named Zurich, a 14 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.
|