Update app.py

app.py

@@ -1,4 +1,4 @@
-import gradio as gr
+"""import gradio as gr
 import nltk
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
@@ -59,3 +59,39 @@ iface = gr.Interface(
 )
 
 iface.launch(share=True)
+"""
+import gradio as gr
+import nltk
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+
+nltk.download('punkt')
+
+def fragment_text(text, tokenizer):
+    # Your existing code for text fragmentation
+
+def summarize_text(text, tokenizer, model):
+    # Your existing code for text summarization
+
+checkpoint = "tclopess/bart_samsum"
+tokenizer = AutoTokenizer.from_pretrained(checkpoint)
+model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint)
+
+def summarize_with_button(text):
+    return summarize_text(text, tokenizer, model)
+
+iface = gr.Interface(
+    fn=summarize_with_button,
+    inputs=gr.Textbox(label="Enter text to summarize:"),
+    outputs=gr.Textbox(label="Summary:"),
+    title="Text Summarizer with Button",
+)
+
+def summarize_text_button():
+    iface.launch(share=True)
+
+gr.Interface(
+    summarize_text_button, 
+    inputs=None, 
+    outputs=None, 
+    title="Click to Summarize"
+).launch()