vishnun commited on
Commit
204b010
1 Parent(s): bdacafc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -3,8 +3,9 @@ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
3
 
4
  st.title("SpellCorrectorT5")
5
  st.markdown('SpellCorrectorT5 is a fine-tuned version of **pre-trained t5-small model** modelled on randomly selected 50000 sentences modified by [imputing random noises/errors](./random_noiser.py) and trained using transformers. It not only looks for _spelling errors but also looks for the semantics_ in the sentence and suggest other possible words for the incorrect word.')
6
- ttokenizer = AutoTokenizer.from_pretrained("./")
7
- tmodel = AutoModelForSeq2SeqLM.from_pretrained("./")
 
8
  form = st.form("T5-form")
9
 
10
  examples = ["I will return it to yu once it is donr",
 
3
 
4
  st.title("SpellCorrectorT5")
5
  st.markdown('SpellCorrectorT5 is a fine-tuned version of **pre-trained t5-small model** modelled on randomly selected 50000 sentences modified by [imputing random noises/errors](./random_noiser.py) and trained using transformers. It not only looks for _spelling errors but also looks for the semantics_ in the sentence and suggest other possible words for the incorrect word.')
6
+ m_name = "vishnun/tinygram"
7
+ ttokenizer = AutoTokenizer.from_pretrained(m_name)
8
+ tmodel = AutoModelForSeq2SeqLM.from_pretrained(m_name)
9
  form = st.form("T5-form")
10
 
11
  examples = ["I will return it to yu once it is donr",