Commit ·
d25f223
1
Parent(s): b867ce6
Update README.md
Browse files
README.md
CHANGED
|
@@ -56,7 +56,11 @@ Please notice that we encourage you to read our tutorials and learn more about
|
|
| 56 |
```python
|
| 57 |
from speechbrain.pretrained import ResponseGenerator
|
| 58 |
res_gen_model = ResponseGenerator.from_hparams(source="speechbrain/MultiWOZ-GPT-Response_Generation", savedir="pretrained_models/MultiWOZ-GPT-Response_Generation", pymodule_file="custom.py")
|
| 59 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 60 |
```
|
| 61 |
|
| 62 |
### Inference on GPU
|
|
|
|
| 56 |
```python
|
| 57 |
from speechbrain.pretrained import ResponseGenerator
|
| 58 |
res_gen_model = ResponseGenerator.from_hparams(source="speechbrain/MultiWOZ-GPT-Response_Generation", savedir="pretrained_models/MultiWOZ-GPT-Response_Generation", pymodule_file="custom.py")
|
| 59 |
+
print("Hi,How could I help you today?", end="\n")
|
| 60 |
+
while True:
|
| 61 |
+
turn = input()
|
| 62 |
+
res_gen_model.generate_response(turn)
|
| 63 |
+
print(response, end="\n")
|
| 64 |
```
|
| 65 |
|
| 66 |
### Inference on GPU
|