Update README.md

README.md

@@ -45,7 +45,6 @@ import torch
 from peft import AutoPeftModelForCausalLM
 from transformers import AutoTokenizer, pipeline
 
-
 # Specify the model ID
 peft_model_id = "Menouar/falcon7b-linear-equations"
 
@@ -62,8 +61,13 @@ pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
 equation = "Solve for y: 10 + 4y -9y +5 = 4 +8y - 2y + 8 ."
 
-outputs = pipe(equation, max_new_tokens=172, do_sample=True, temperature=0.1, top_k=50, top_p=0.1,
- eos_token_id=pipe.tokenizer.eos_token_id, pad_token_id=pipe.tokenizer.pad_token_id)
+outputs = pipe(equation, 
+ max_new_tokens=172, 
+ do_sample=True, 
+ temperature=0.1,
+ top_k=50, top_p=0.1,
+ eos_token_id=pipe.tokenizer.eos_token_id,
+ pad_token_id=pipe.tokenizer.pad_token_id)
 
 for seq in outputs:
  print(f"{seq['generated_text']}")