poonehmousavi
commited on
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
|