Update app.py

app.py

@@ -99,6 +99,8 @@ if st.button("Process",type="primary"):
                 summary = []
 
                 for x in stqdm(range(len(text))):
+                     if st.button("Cancel"):
+                        break
                     try:
                         summary.append(
                             model(
@@ -143,7 +145,10 @@ if st.button("Process",type="primary"):
                 model, tokenizer = load_t5()
                 summary = []
                 for x in stqdm(range(len(text))):
-
+                    if st.button("Cancel"):
+                        break
+                    
+            
                     tokens_input = tokenizer.encode(
                         "summarize: " + text[x],
                         return_tensors="pt",
@@ -199,6 +204,8 @@ if st.button("Process",type="primary"):
 
                 summary = []
                 for x in stqdm(range(len(text))):
+                     if st.button("Cancel"):
+                        break
                     try:
                         summary.append(model.predict(text[x])[0])
                     except: