Update README.md

README.md

@@ -99,7 +99,7 @@ inputs = tokenizer('''def print_prime(n):
    """''', return_tensors="pt", return_attention_mask=False)
 
 outputs = model.generate(**inputs, max_length=200)
-text = tokenizer.batch_decode(outputs[0])
+text = tokenizer.batch_decode(outputs)[0]
 print(text)
 ```