Update README.md

README.md

@@ -126,7 +126,7 @@ prediction = model.generate(
     temperature=0.7,
     num_return_sequences=1,
 )
-prediction_text = tokenizer.decode(prediction[0])
+prediction_text = tokenizer.decode(prediction[0].tolist())
 print(prediction_text)
 ```