Update README.md

README.md

@@ -44,7 +44,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id).to("cuda")
 
 
-def create_instruction(instruction: str, input_data: str = None, context: str = None) -> str:
+def create_instruction(instruction, input_data=None, context=None):
     sections = {
         "Instrucción": instruction,
         "Entrada": input_data,
@@ -63,7 +63,6 @@ def create_instruction(instruction: str, input_data: str = None, context: str =
     return prompt
 
 
-
 def generate(
         instruction,
         input=None,