Update README.md

README.md

@@ -20,11 +20,11 @@ widget:
 - text: "Uma conferência da ONU terminou no início do sábado com um plano vago para novas negociações informais sobre como diminuir o aquecimento global, mas sem um compromisso dos EUA com negociações multilaterais sobre os próximos passos, incluindo controles de emissões."
   example_title: Exemplo
 ---
-# TeenyTinyLlama-162m-AgNews
+# TeenyTinyLlama-160m-AgNews
 
 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-AgNews`) fine-tuned on the [AgNews dataset](https://huggingface.co/datasets/maritaca-ai/ag_news_pt).
+This repository contains a version of [TeenyTinyLlama-160m](https://huggingface.co/nicholasKluge/TeenyTinyLlama-160m) (`TeenyTinyLlama-160m-AgNews`) fine-tuned on the [AgNews dataset](https://huggingface.co/datasets/maritaca-ai/ag_news_pt).
 
 ## Details
 
@@ -43,7 +43,7 @@ from transformers import pipeline
 text = "Quando falamos de Inteligência Artificial, hoje em dia existem já vários sistemas que começam a ganhar popularidade,\
  embora nenhum seja ainda tão conhecido como o ChatGPT da OpenAI."
 
-classifier = pipeline("text-classification", model="nicholasKluge/TeenyTinyLlama-162m-AgNews")
+classifier = pipeline("text-classification", model="nicholasKluge/TeenyTinyLlama-160m-AgNews")
 classifier(text)
 
 # >>> [{'label': 'TECNOLOGIA', 'score': 0.9997298121452332}] 
@@ -69,13 +69,13 @@ dataset = load_dataset("maritaca-ai/ag_news_pt")
 
 # Create a `ModelForSequenceClassification`
 model = AutoModelForSequenceClassification.from_pretrained(
-    "nicholasKluge/TeenyTinyLlama-162m",
+    "nicholasKluge/TeenyTinyLlama-160m",
     num_labels=4,
     id2label={0: "MUNDO", 1: "ESPORTES", 2: "NEGÓCIOS", 3: "TECNOLOGIA"},
     label2id={"MUNDO": 0, "ESPORTES": 1, "NEGÓCIOS": 2, "TECNOLOGIA": 3}
 )
 
-tokenizer = AutoTokenizer.from_pretrained("nicholasKluge/TeenyTinyLlama-162m")
+tokenizer = AutoTokenizer.from_pretrained("nicholasKluge/TeenyTinyLlama-160m")
 
 # Format the dataset
 train = dataset['train'].to_pandas()
@@ -147,7 +147,7 @@ trainer.train()
 
 | Models                                                                                     | [AgNews](https://huggingface.co/datasets/maritaca-ai/ag_news_pt) | 
 |--------------------------------------------------------------------------------------------|------------------------------------------------------------------|
-| [Teeny Tiny Llama 162m](https://huggingface.co/nicholasKluge/TeenyTinyLlama-162m)          | 94.05                                                            | 
+| [Teeny Tiny Llama 160m](https://huggingface.co/nicholasKluge/TeenyTinyLlama-160m)          | 94.05                                                            | 
 | [Bert-base-portuguese-cased](https://huggingface.co/neuralmind/bert-base-portuguese-cased) | 94.19                                                            |
 | [Bert-large-portuguese-cased](https://huggingface.co/neuralmind/bert-base-portuguese-cased)| 94.11                                                            |
 | [Gpt2-small-portuguese](https://huggingface.co/pierreguillou/gpt2-small-portuguese)        | 94.07                                                            |
@@ -158,7 +158,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},
@@ -174,4 +174,4 @@ This repository was built as part of the RAIES ([Rede de Inteligência Artificia
 
 ## License
 
-TeenyTinyLlama-162m-AgNews is licensed under the Apache License, Version 2.0. See the [LICENSE](LICENSE) file for more details.
+TeenyTinyLlama-160m-AgNews is licensed under the Apache License, Version 2.0. See the [LICENSE](LICENSE) file for more details.