Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ import spaces
|
|
3 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
4 |
import torch
|
5 |
|
6 |
-
model_name = "rubenroy/Zurich-
|
7 |
model = AutoModelForCausalLM.from_pretrained(
|
8 |
model_name,
|
9 |
torch_dtype=torch.bfloat16,
|
@@ -115,7 +115,7 @@ TITLE_HTML = """
|
|
115 |
<p style="font-size: 1.25rem; color: #94a3b8; margin: 0;">GammaCorpus v2-5m</p>
|
116 |
</div>
|
117 |
<div class="info-section">
|
118 |
-
<span>Fine-tuned from <a href="https://huggingface.co/Qwen/Qwen2.5-7B-Instruct" class="info-link">Qwen 2.5 7B Instruct</a> | Model: <a href="https://huggingface.co/rubenroy/Zurich-
|
119 |
</div>
|
120 |
</div>
|
121 |
|
|
|
3 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
4 |
import torch
|
5 |
|
6 |
+
model_name = "rubenroy/Zurich-7B-GCv2-5m"
|
7 |
model = AutoModelForCausalLM.from_pretrained(
|
8 |
model_name,
|
9 |
torch_dtype=torch.bfloat16,
|
|
|
115 |
<p style="font-size: 1.25rem; color: #94a3b8; margin: 0;">GammaCorpus v2-5m</p>
|
116 |
</div>
|
117 |
<div class="info-section">
|
118 |
+
<span>Fine-tuned from <a href="https://huggingface.co/Qwen/Qwen2.5-7B-Instruct" class="info-link">Qwen 2.5 7B Instruct</a> | Model: <a href="https://huggingface.co/rubenroy/Zurich-7B-GCv2-5m" class="info-link">Zurich-7B-GCv2-5m</a> | Training Dataset: <a href="https://huggingface.co/datasets/rubenroy/GammaCorpus-v2-5m" class="info-link">GammaCorpus v2 5m</a></span>
|
119 |
</div>
|
120 |
</div>
|
121 |
|