Update app.py

app.py

@@ -70,16 +70,28 @@ if button:
     if text:
         if model == "Transformer":
             st.write("You have selected Transformer model.")
-            summary = transform_summarize(text)
+            try:
+                summary = transform_summarize(text)
+            except Exception:
+                st.warning("🚨 Your input text is quite lengthy. For better results, consider providing a shorter text or breaking it into smaller chunks.")
         elif model == "T5":
             st.write("You have selected T5 model.")
-            summary = t5_summarize(text)
+            try:
+                summary = t5_summarize(text)
+            except Exception:
+                st.warning("🚨 Your input text is quite lengthy. For better results, consider providing a shorter text or breaking it into smaller chunks.")
         elif model == "BART":
             st.write("You have selected BART model.")
-            summary = bart_summarize(text)
+            try:
+                summary = bart_summarize(text)
+            except Exception:
+                st.warning("🚨 Your input text is quite lengthy. For better results, consider providing a shorter text or breaking it into smaller chunks.")
         elif model == "Encoder-Decoder":
             st.write("You have selected Encoder-Decoder model.")
-            summary = encoder_decoder(text)
+            try:
+                summary = encoder_decoder(text)
+            except Exception:
+                st.warning("🚨 Your input text is quite lengthy. For better results, consider providing a shorter text or breaking it into smaller chunks.")
 
         #st.toast("Please wait while we summarize your text.")
         #with st.spinner("Summarizing..."):