Update README.md

README.md

@@ -71,17 +71,17 @@ model.eval()
 # Based on the inference code by `tloen/alpaca-lora`
 def generate_prompt(instruction, input=None):
     if input:
-        return f"""Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.
-### Instruction:
+        return f"""A continuación se muestra una instrucción que describe una tarea, emparejada con una entrada que proporciona más contexto. Escribe una respuesta que complete adecuadamente la petición.
+### Instrucción:
 {instruction}
-### Input:
+### Entrada:
 {input}
-### Response:"""
+### Respuesta:"""
     else:
-        return f"""Below is an instruction that describes a task. Write a response that appropriately completes the request.
-### Instruction:
+        return f"""A continuación se muestra una instrucción que describe una tarea. Escribe una respuesta que complete adecuadamente la petición.
+### Instrucción:
 {instruction}
-### Response:"""
+### Respuesta:"""
 
 def generate(
         instruction,
@@ -112,10 +112,10 @@ def generate(
         )
     s = generation_output.sequences[0]
     output = tokenizer.decode(s)
-    return output.split("### Response:")[1].strip().split("Below")[0]
+    return output.split("### Response:")[1]
 
 instruction = "¿Qué es un chivo?"
 
-print("Instruction:", instruction)
-print("Response:", generate(instruction))
+print("Instrucción:", instruction)
+print("Respuesta:", generate(instruction))
 ```