Update README.md

README.md

@@ -62,10 +62,14 @@ encoding = tokenizer(ytb['Title'][i], return_tensors="pt")
 input_ids, attention_masks = encoding["input_ids"].to("cuda"), encoding["attention_mask"].to("cuda")
 
 outputs = model.generate(
+
     input_ids=input_ids, attention_mask=attention_masks,
+    
     max_length=30,
+    
     early_stopping=True
 )
 
 for output in outputs:
+
   outputs = tokenizer.decode(output, skip_special_tokens=True, clean_up_tokenization_spaces=True)