Spaces:
Running
Running
Commit
·
beb7c02
1
Parent(s):
37055d8
added one close button :-)
Browse files
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(
|