Update README.md
Browse files
README.md
CHANGED
@@ -6,23 +6,26 @@ license: mit
|
|
6 |
---
|
7 |
An generative AI made using [microsoft/DialoGPT-small](https://huggingface.co/microsoft/DialoGPT-small).
|
8 |
|
9 |
-
Trained on
|
10 |
|
11 |
-
https://www.kaggle.com/Cornell-University/movie-dialog-corpus
|
12 |
|
13 |
-
https://www.kaggle.com/jef1056/discord-data
|
14 |
|
15 |
|
16 |
|
17 |
-
Important
|
18 |
-
|
19 |
-
|
20 |
-
|
21 |
-
|
22 |
|
23 |
-
|
|
|
|
|
|
|
24 |
|
25 |
-
Example
|
26 |
```python
|
27 |
from transformers import AutoTokenizer, AutoModelWithLMHead
|
28 |
|
|
|
6 |
---
|
7 |
An generative AI made using [microsoft/DialoGPT-small](https://huggingface.co/microsoft/DialoGPT-small).
|
8 |
|
9 |
+
Trained on:
|
10 |
|
11 |
+
https://www.kaggle.com/Cornell-University/movie-dialog-corpus
|
12 |
|
13 |
+
https://www.kaggle.com/jef1056/discord-data
|
14 |
|
15 |
|
16 |
|
17 |
+
Important:
|
18 |
+
The AI can be a bit weird at times as it is still undergoing training!
|
19 |
+
At times it send stuff using :<random_wierd_words>: as they are discord emotes.
|
20 |
+
It also send random @RandomName as it is trying to ping people.
|
21 |
+
This works well on discord but on the web not so much but it is easy enough to remove such stuff using [re.sub](https://docs.python.org/3/library/re.html#re.sub)
|
22 |
|
23 |
+
|
24 |
+
|
25 |
+
Issues:
|
26 |
+
The AI like with all conversation AI lacks a character, it changes its name way too often. This can be solved using an AIML chatbot to give it a stable character!
|
27 |
|
28 |
+
Example:
|
29 |
```python
|
30 |
from transformers import AutoTokenizer, AutoModelWithLMHead
|
31 |
|