Update README.md

README.md

@@ -17,6 +17,7 @@ TeenyTinyLlama is a series of small foundational models trained on Portuguese.
 
 This repository contains a version of [TeenyTinyLlama-162m]() fine-tuned on a translated version of the IMDB dataset.
 
+## Reproducing
   
 ```python
 # IMDB
@@ -26,12 +27,7 @@ import evaluate
 import numpy as np
 from datasets import load_dataset, Dataset, DatasetDict
 from transformers import AutoTokenizer, DataCollatorWithPadding
-from transformers import AutoModelForSequenceClassification, TrainingArguments, Trainer
-
-
-evaluation_strategy="epoch"
-save_strategy="epoch"
-hub_model_id="nicholasKluge/Teeny-tiny-llama-162m-imdb"
+from transformers import AutoModelForSequenceClassification, TrainingArguments, Traine
 
 # Load the task
 dataset = load_dataset("christykoh/imdb_pt")
@@ -72,14 +68,12 @@ training_args = TrainingArguments(
     per_device_eval_batch_size=16,
     num_train_epochs=3,
     weight_decay=0.01,
-    evaluation_strategy=evaluation_strategy,
-    save_strategy=save_strategy,
+    evaluation_strategy="epoch",
+    save_strategy="epoch",
     load_best_model_at_end=True,
-    push_to_hub=False,
-    hub_token=token,
-    hub_private_repo=True,
-    hub_model_id=hub_model_id,
-    tf32=False,
+    push_to_hub=True,
+    hub_token="your_token_here",
+    hub_model_id="username/model-name-imdb
 )
 
 # Define the Trainer