Update README.md

README.md

@@ -57,20 +57,20 @@ def generate_prompt(instruction, input=None):
     if input:
         return f"""Abaixo está uma instrução que descreve uma tarefa, emparelhada com uma entrada que fornece mais contexto. Escreva uma resposta que conclua adequadamente a solicitação.
 
-### Instruções:
+### Instruction:
 {instruction}
 
-### Entrada:
+### Input:
 {input}
 
-### Resposta:"""
+### Response:"""
     else:
         return f"""Abaixo está uma instrução que descreve uma tarefa. Escreva uma resposta que conclua adequadamente a solicitação.
 
-### Instruções:
+### Instruction:
 {instruction}
 
-### Resposta:"""
+### Response:"""
 
 generation_config = GenerationConfig(
     temperature=0.1,
@@ -91,7 +91,7 @@ def generate(instruction, input=None):
     )
     for s in generation_output.sequences:
         output = tokenizer.decode(s)
-        print("Resposta:", output.split("### Resposta:")[1].strip())
+        print("Resposta:", output.split("### Response:")[1].strip())
 
 generate(input("Instrução: "))