Update app.py
Browse files
app.py
CHANGED
|
@@ -198,5 +198,7 @@ def main():
|
|
| 198 |
summary_ids = model.generate(inputs, max_length=150, min_length=80, length_penalty=5., num_beams=2)
|
| 199 |
summary = tokenizer.decode(summary_ids[0])
|
| 200 |
st.success(summary)
|
|
|
|
|
|
|
| 201 |
if __name__ == '__main__':
|
| 202 |
main()
|
|
|
|
| 198 |
summary_ids = model.generate(inputs, max_length=150, min_length=80, length_penalty=5., num_beams=2)
|
| 199 |
summary = tokenizer.decode(summary_ids[0])
|
| 200 |
st.success(summary)
|
| 201 |
+
if st.button("refresh"):
|
| 202 |
+
st.experimental_rerun()
|
| 203 |
if __name__ == '__main__':
|
| 204 |
main()
|