Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 import torch
+import sys
 
 def generate_summary(model, tokenizer, dialogue):
     # Tokenize input dialogue
@@ -21,7 +22,7 @@ st.title("Dialog Summarizer App")
 user_input = st.text_area("Enter the dialog:")
 if not user_input:
     st.info("Please enter a dialog.")
-    return
+    sys.exit()
 
 # Load pre-trained Pegasus model and tokenizer
 model_name = "ale-dp/pegasus-finetuned-dialog-summarizer"