Spaces:
Sleeping
Sleeping
Update README.md
Browse files
README.md
CHANGED
|
@@ -99,4 +99,12 @@ For training the model to transfer style of Rachel to chatbot I used several mod
|
|
| 99 |
1. First one is GPT2. For datasets I used TextDataset by PyTorch and library transformers by huggingface.
|
| 100 |
The results are shown on the image below
|
| 101 |

|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 102 |
|
|
|
|
| 99 |
1. First one is GPT2. For datasets I used TextDataset by PyTorch and library transformers by huggingface.
|
| 100 |
The results are shown on the image below
|
| 101 |

|
| 102 |
+
|
| 103 |
+
2. Second model is GPT2-medium. Training on monologues is shown below
|
| 104 |
+

|
| 105 |
+
Training on dialogues is shown on the next image
|
| 106 |
+

|
| 107 |
+
Result of the training is shown below
|
| 108 |
+

|
| 109 |
+
|
| 110 |
|