vishnun commited on
Commit
6afb4da
1 Parent(s): 403d5d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import streamlit as st
2
  from transformers import T5ForConditionalGeneration, AutoTokenizer
3
 
4
- st.title("SimpleGrammarlyT5")
5
- st.markdown('SimpleGrammarlyT5 is a fine-tuned version of **pre-trained t5-small model** modelled on randomly selected 50000 sentences modified by imputing random noises/errors 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 = T5ForConditionalGeneration.from_pretrained('./')
8
 
@@ -19,7 +19,7 @@ if submit:
19
  do_sample=True,
20
  max_length=50,
21
  top_p=0.98,
22
- num_return_sequences=3
23
  )
24
 
25
  st.subheader("Suggested sentences: ")
 
1
  import streamlit as st
2
  from transformers import T5ForConditionalGeneration, 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 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 = T5ForConditionalGeneration.from_pretrained('./')
8
 
 
19
  do_sample=True,
20
  max_length=50,
21
  top_p=0.98,
22
+ num_return_sequences=2
23
  )
24
 
25
  st.subheader("Suggested sentences: ")