Spaces:
Runtime error
Runtime error
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(
|