ashhadahsan commited on
Commit
beb7c02
·
1 Parent(s): 37055d8

added one close button :-)

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def load_t5():
35
 
36
  tokenizer = AutoTokenizer.from_pretrained("t5-base")
37
  return model, tokenizer
38
-
39
 
40
  @st.cache
41
  def custom_model():
@@ -99,6 +99,9 @@ if st.button("Process",type="primary"):
99
  summary = []
100
 
101
  for x in stqdm(range(len(text))):
 
 
 
102
  try:
103
  summary.append(
104
  model(
 
35
 
36
  tokenizer = AutoTokenizer.from_pretrained("t5-base")
37
  return model, tokenizer
38
+ cancel_button=st.empty()
39
 
40
  @st.cache
41
  def custom_model():
 
99
  summary = []
100
 
101
  for x in stqdm(range(len(text))):
102
+
103
+ if cancel_button.button("Cancel",key=x):
104
+ break
105
  try:
106
  summary.append(
107
  model(