diversifix commited on
Commit
f4ad5df
1 Parent(s): 993815d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -1
README.md CHANGED
@@ -31,7 +31,7 @@ An experimental model version is released [on Huggingface](https://huggingface.c
31
  ```python
32
  from transformers import T5Tokenizer, TFT5ForConditionalGeneration
33
 
34
- tokenizer = T5Tokenizer.from_pretrained("google/mt5-small")
35
  model = TFT5ForConditionalGeneration.from_pretrained("diversifix/diversiformer")
36
 
37
  def generate(prompt, tokenizer, model):
@@ -50,6 +50,10 @@ for prompt in prompts:
50
  print(f"{prompt}\n{output}\n\n")
51
  ```
52
 
 
 
 
 
53
  ## License
54
 
55
  Diversiformer. Transformer model for inclusive language.
 
31
  ```python
32
  from transformers import T5Tokenizer, TFT5ForConditionalGeneration
33
 
34
+ tokenizer = T5Tokenizer.from_pretrained("diversifix/diversiformer")
35
  model = TFT5ForConditionalGeneration.from_pretrained("diversifix/diversiformer")
36
 
37
  def generate(prompt, tokenizer, model):
 
50
  print(f"{prompt}\n{output}\n\n")
51
  ```
52
 
53
+ ## Dev ideas
54
+
55
+ - Use classifier to filter out training data of low quality? (~adversarial approach)
56
+
57
  ## License
58
 
59
  Diversiformer. Transformer model for inclusive language.