Soumen commited on
Commit
b59b45a
·
1 Parent(s): 7e1ea57

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -111,6 +111,8 @@ def entity_analyzer(my_text):
111
  allData = ['"Token":{},\n"Entities":{}'.format(tokens,entities)]
112
  return allData
113
  def main():
 
 
114
  """ NLP Based Application with Streamlit """
115
  st.markdown("""
116
  #### Description
@@ -192,8 +194,6 @@ def main():
192
  # st.success(summary_result)
193
  if st.checkbox("Mark to English Text Summarization!"):
194
  #st.title("Summarize Your Text for English only!")
195
- tokenizer = AutoTokenizer.from_pretrained('t5-base')
196
- model = AutoModelWithLMHead.from_pretrained('t5-base', return_dict=True)
197
  #st.text("Using Google T5 Transformer ..")
198
  inputs = tokenizer.encode("summarize: " + text,
199
  return_tensors='pt',
@@ -202,7 +202,7 @@ def main():
202
  summary_ids = model.generate(inputs, max_length=150, min_length=80, length_penalty=5., num_beams=2)
203
  summary = tokenizer.decode(summary_ids[0])
204
  st.success(summary)
205
- if st.button("refresh"):
206
- st.experimental_rerun()
207
  if __name__ == '__main__':
208
  main()
 
111
  allData = ['"Token":{},\n"Entities":{}'.format(tokens,entities)]
112
  return allData
113
  def main():
114
+ tokenizer = AutoTokenizer.from_pretrained('t5-base')
115
+ model = AutoModelWithLMHead.from_pretrained('t5-base', return_dict=True)
116
  """ NLP Based Application with Streamlit """
117
  st.markdown("""
118
  #### Description
 
194
  # st.success(summary_result)
195
  if st.checkbox("Mark to English Text Summarization!"):
196
  #st.title("Summarize Your Text for English only!")
 
 
197
  #st.text("Using Google T5 Transformer ..")
198
  inputs = tokenizer.encode("summarize: " + text,
199
  return_tensors='pt',
 
202
  summary_ids = model.generate(inputs, max_length=150, min_length=80, length_penalty=5., num_beams=2)
203
  summary = tokenizer.decode(summary_ids[0])
204
  st.success(summary)
205
+ if st.button("refresh"):
206
+ st.experimental_rerun()
207
  if __name__ == '__main__':
208
  main()