oluwatosin adewumi commited on
Commit
28ed5a4
1 Parent(s): 5187dc7

readme doc edited

Browse files
Files changed (1) hide show
  1. README.md +15 -6
README.md CHANGED
@@ -14,23 +14,32 @@ widget:
14
  - text: "How I fit chop for here?"
15
  ---
16
 
17
- ## DialoGPT_AfriWOZ
 
18
  This is a fine-tuned model of DialoGPT (small) on the AfriWOZ dataset. It is intended to be used as a conversational system in Nigeria Pidgin English language.
19
  The dataset it's trained on is limited in scope, as it covers only certain domains such as restaurants, hotel, taxi, and booking.
20
- The perplexity achieved on the validation set 38.52.
 
 
21
  * Generation example from an interactive environment:
 
22
  |Role | Response |
23
- |---------|------------|
24
  |User | I hear say restaurant dey here. |
25
- |Bot | |
26
  |User | Abeg you fit tell me which kind chop dey? |
27
- |Bot | |
28
  |User | You do well. Thank you. |
29
- |Bot | |
 
30
  Please find the information about preprocessing, training and full details of the DialoGPT in the [original DialoGPT repository](https://github.com/microsoft/DialoGPT)
 
31
  The paper for this work can be found on arXiv: [https://arxiv.org/pdf/2204.08083.pdf](https://arxiv.org/pdf/2204.08083.pdf)
 
32
  ### How to use
 
33
  Now we are ready to try out how the model works as a chatting partner!
 
34
  ```python
35
  from transformers import AutoModelForCausalLM, AutoTokenizer
36
  import torch
 
14
  - text: "How I fit chop for here?"
15
  ---
16
 
17
+ ## DialoGPT_AfriWOZ (Pidgin)
18
+
19
  This is a fine-tuned model of DialoGPT (small) on the AfriWOZ dataset. It is intended to be used as a conversational system in Nigeria Pidgin English language.
20
  The dataset it's trained on is limited in scope, as it covers only certain domains such as restaurants, hotel, taxi, and booking.
21
+
22
+ The perplexity achieved on the validation set is 38.52.
23
+
24
  * Generation example from an interactive environment:
25
+
26
  |Role | Response |
27
+ |------|------------|
28
  |User | I hear say restaurant dey here. |
29
+ |Bot | I dey find restaurant. |
30
  |User | Abeg you fit tell me which kind chop dey? |
31
+ |Bot | I no dey sure right now. |
32
  |User | You do well. Thank you. |
33
+ |Bot | I dey find restauarant. |
34
+
35
  Please find the information about preprocessing, training and full details of the DialoGPT in the [original DialoGPT repository](https://github.com/microsoft/DialoGPT)
36
+
37
  The paper for this work can be found on arXiv: [https://arxiv.org/pdf/2204.08083.pdf](https://arxiv.org/pdf/2204.08083.pdf)
38
+
39
  ### How to use
40
+
41
  Now we are ready to try out how the model works as a chatting partner!
42
+
43
  ```python
44
  from transformers import AutoModelForCausalLM, AutoTokenizer
45
  import torch