sino commited on
Commit ·
c96ae54
1
Parent(s): c458959
Update README.md
Browse files
README.md
CHANGED
|
@@ -73,7 +73,11 @@ input['spectrogram'] = torch.from_numpy(lms).unsqueez(dim=0).to(device)
|
|
| 73 |
model.eval()
|
| 74 |
gen_ids = model.forward_test(input)
|
| 75 |
gen_text = model.neck.tokenizer.batch_decode(gen_ids.clip(0))
|
|
|
|
|
|
|
|
|
|
| 76 |
|
|
|
|
| 77 |
```
|
| 78 |
|
| 79 |
|
|
|
|
| 73 |
model.eval()
|
| 74 |
gen_ids = model.forward_test(input)
|
| 75 |
gen_text = model.neck.tokenizer.batch_decode(gen_ids.clip(0))
|
| 76 |
+
gen_text = gen_text.split('<s>')[-1].split('\n')[0].strip()
|
| 77 |
+
gen_text = gen_text.replace(' in Chinese', '')
|
| 78 |
+
gen_text = gen_text.replace(' Chinese', '')
|
| 79 |
|
| 80 |
+
print(gen_text)
|
| 81 |
```
|
| 82 |
|
| 83 |
|