Update README.md
Browse files
README.md
CHANGED
|
@@ -80,9 +80,6 @@ input_dic['spectrogram'] = torch.from_numpy(lms).unsqueez(dim=0).to(device)
|
|
| 80 |
model.eval()
|
| 81 |
gen_ids = model.forward_test(input)
|
| 82 |
gen_text = model.neck.tokenizer.batch_decode(gen_ids.clip(0))
|
| 83 |
-
gen_text = gen_text.split('<s>')[-1].split('\n')[0].strip()
|
| 84 |
-
gen_text = gen_text.replace(' in Chinese', '')
|
| 85 |
-
gen_text = gen_text.replace(' Chinese', '')
|
| 86 |
|
| 87 |
print(gen_text)
|
| 88 |
```
|
|
|
|
| 80 |
model.eval()
|
| 81 |
gen_ids = model.forward_test(input)
|
| 82 |
gen_text = model.neck.tokenizer.batch_decode(gen_ids.clip(0))
|
|
|
|
|
|
|
|
|
|
| 83 |
|
| 84 |
print(gen_text)
|
| 85 |
```
|