Update README.md

README.md

@@ -16,11 +16,11 @@ widget:
 - text: "<s>Uma mulher está misturando ovos.<s>A mulher está bebendo.</s>"
   example_title: Exemplo
 ---
-# TeenyTinyLlama-162m-Assin2
+# TeenyTinyLlama-160m-Assin2
 
 TeenyTinyLlama is a series of small foundational models trained in Brazilian Portuguese.
 
-This repository contains a version of [TeenyTinyLlama-162m](https://huggingface.co/nicholasKluge/TeenyTinyLlama-162m) (`TeenyTinyLlama-162m-Assin2`) fine-tuned on the [Assin2](https://huggingface.co/datasets/assin2).
+This repository contains a version of [TeenyTinyLlama-160m](https://huggingface.co/nicholasKluge/TeenyTinyLlama-160m) (`TeenyTinyLlama-160m-Assin2`) fine-tuned on the [Assin2](https://huggingface.co/datasets/assin2).
 
 ## Details
 
@@ -38,7 +38,7 @@ from transformers import pipeline
 
 text = "<s>Qual a capital do Brasil?<s>A capital do Brasil é Brasília!</s>"
 
-classifier = pipeline("text-classification", model="nicholasKluge/TeenyTinyLlama-162m-Assin2")
+classifier = pipeline("text-classification", model="nicholasKluge/TeenyTinyLlama-160m-Assin2")
 classifier(text)
 
 # >>> [{'label': 'ENTAILED', 'score': 0.9392824769020081}]
@@ -63,13 +63,13 @@ dataset = load_dataset("assin2")
 
 # Create a `ModelForSequenceClassification`
 model = AutoModelForSequenceClassification.from_pretrained(
-    "nicholasKluge/TeenyTinyLlama-162m", 
+    "nicholasKluge/TeenyTinyLlama-160m", 
     num_labels=2, 
     id2label={0: "UNENTAILED", 1: "ENTAILED"}, 
     label2id={"UNENTAILED": 0, "ENTAILED": 1}
 )
 
-tokenizer = AutoTokenizer.from_pretrained("nicholasKluge/TeenyTinyLlama-162m")
+tokenizer = AutoTokenizer.from_pretrained("nicholasKluge/TeenyTinyLlama-160m")
 
 # Format the dataset
 train = dataset['train'].to_pandas()
@@ -146,7 +146,7 @@ trainer.train()
 
 | Models                                                                                     | [Assin2](https://huggingface.co/datasets/assin2)|
 |--------------------------------------------------------------------------------------------|-------------------------------------------------|
-| [Teeny Tiny Llama 162m](https://huggingface.co/nicholasKluge/TeenyTinyLlama-162m)          | 85.78                                           |
+| [Teeny Tiny Llama 160m](https://huggingface.co/nicholasKluge/TeenyTinyLlama-160m)          | 85.78                                           |
 | [Bert-base-portuguese-cased](https://huggingface.co/neuralmind/bert-base-portuguese-cased) | 87.45                                           |
 | [Bert-large-portuguese-cased](https://huggingface.co/neuralmind/bert-base-portuguese-cased)| 88.97                                           |
 | [Gpt2-small-portuguese](https://huggingface.co/pierreguillou/gpt2-small-portuguese)        | 86.11                                           |
@@ -157,7 +157,7 @@ trainer.train()
 
 @misc{nicholas22llama,
   doi = {10.5281/zenodo.6989727},
-  url = {https://huggingface.co/nicholasKluge/TeenyTinyLlama-162m},
+  url = {https://huggingface.co/nicholasKluge/TeenyTinyLlama-160m},
   author = {Nicholas Kluge Corrêa},
   title = {TeenyTinyLlama},
   year = {2023},
@@ -173,4 +173,4 @@ This repository was built as part of the RAIES ([Rede de Inteligência Artificia
 
 ## License
 
-TeenyTinyLlama-162m-Assin2 is licensed under the Apache License, Version 2.0. See the [LICENSE](LICENSE) file for more details.
+TeenyTinyLlama-160m-Assin2 is licensed under the Apache License, Version 2.0. See the [LICENSE](LICENSE) file for more details.