poonehmousavi commited on
Commit
d25f223
1 Parent(s): b867ce6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -1
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
- res_gen_model.generate_response()
 
 
 
 
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