speechbrain
/

MultiWOZ-GPT-Response_Generation

response-generation

Model card Files Files and versions Community

poonehmousavi commited on Dec 14, 2023

Commit

a8688f4

•

1 Parent(s): 4483827

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -57,8 +57,8 @@ Please notice that we encourage you to read our tutorials and learn more about
 ### Generating your Own Dialuge
 ```python
-from speechbrain.pretrained import ResponseGenerator
-res_gen_model = ResponseGenerator.from_hparams(source="speechbrain/MultiWOZ-GPT-Response_Generation", savedir="pretrained_models/MultiWOZ-GPT-Response_Generation", pymodule_file="custom.py")
 print("Hi,How could I help you today?", end="\n")
 while True:
   turn = input()
@@ -96,7 +96,7 @@ pip install -e .
 3. Run Training:
 ```
-cd recipes/MultiWOZ/response_generation
 pip install -r extra_requirements.txt
 python train_with_gpt.py hparams/train_gpt.yaml --data_folder=your_data_folder
 ```

 ### Generating your Own Dialuge
 ```python
+from speechbrain.inference.text import import GPTResponseGenerator
+res_gen_model = GPTResponseGenerator.from_hparams(source="speechbrain/MultiWOZ-GPT-Response_Generation", savedir="pretrained_models/MultiWOZ-GPT-Response_Generation", pymodule_file="custom.py")
 print("Hi,How could I help you today?", end="\n")
 while True:
   turn = input()
 3. Run Training:
 ```
+cd recipes/MultiWOZ/response_generation/gpt/
 pip install -r extra_requirements.txt
 python train_with_gpt.py hparams/train_gpt.yaml --data_folder=your_data_folder
 ```