Update README.md

README.md

@@ -67,8 +67,8 @@ prompt =  f"Translate this from {lang} to Phones:\n{lang}: "
 
 inputs = tokenizer(f"{prompt}{sentence}\nPhones:", return_tensors="pt").to("cuda")
 
-outputs = model.generate(**inputs, max_new_tokens=256, eos_token_id=stop_tokens)
-phonemized = tokenizer.decode(outputs[0], skip_special_tokens=False)
+outputs = model.generate(**inputs, max_new_tokens=256)
+phonemized = tokenizer.decode(outputs[0], skip_special_tokens=True)
 phonemized = phonemized.split("\n")[-1].replace("Phones:", "")
 
 print(phonemized)