oluwatosin adewumi commited on
Commit
f5c5a53
1 Parent(s): 1f69796

Wolof model modified.

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -43,8 +43,8 @@ Now we are ready to try out how the model works as a chatting partner!
43
  ```python
44
  from transformers import AutoModelForCausalLM, AutoTokenizer
45
  import torch
46
- tokenizer = AutoTokenizer.from_pretrained("tosin/dialogpt_mwoz")
47
- model = AutoModelForCausalLM.from_pretrained("tosin/dialogpt_mwoz")
48
  # Let's chat for 5 lines
49
  for step in range(5):
50
  # encode the new user input, add the eos_token and return a tensor in Pytorch
 
43
  ```python
44
  from transformers import AutoModelForCausalLM, AutoTokenizer
45
  import torch
46
+ tokenizer = AutoTokenizer.from_pretrained("tosin/dialogpt_afriwoz_wolof")
47
+ model = AutoModelForCausalLM.from_pretrained("tosin/dialogpt_afriwoz_wolof")
48
  # Let's chat for 5 lines
49
  for step in range(5):
50
  # encode the new user input, add the eos_token and return a tensor in Pytorch