Update README.md

README.md

@@ -10,8 +10,8 @@ language:
 # Modelo Llama 1 para Tiny Shakespeare
 
 Este repositório contém um modelo de linguagem Llama 1 com 2.3 milhões de parâmetros treinado no dataset Tiny Shakespeare. O modelo foi projetado para tarefas de geração de texto, em especial para o estilo de escrita do Shakespeare.
-Vale ressaltar que existem dois modelos neste repositório. O `llama1-model.pt` é o modelo obtido ao final do treinamento, e o `best-model.pt` é o modelo com os melhores valores das métricas de acurácia e perplexidade. 
-Ambos foram testados e o desempenho foi bem similar, logo, é mais conveniente utilizar o `llama1-model.pt`, pois ele foi treinado até a condição de parada ser atingida.
+Vale ressaltar que existem dois modelos neste repositório. O `llama1-model.pt` é o modelo obtido ao final do treinamento, e o `best_model.pt` é o modelo com os melhores valores das métricas de acurácia e perplexidade. 
+Ambos foram testados e o best-model faz a geração de textos mais coerentes, logo, é mais conveniente utilizar o `best-model.pt`.
 
 ## Arquitetura
 
@@ -51,4 +51,4 @@ Para carregar e utilizar o modelo:
 import torch
 
 # Carregue o modelo
-model = torch.load("llama1-model")
+model = torch.load("best_model.pt")