Update app.py
Browse files
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 |
-
|
206 |
-
|
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()
|