Upload 3 files

README.md

@@ -1,14 +1,26 @@
----
-title: Style Summarizer
-emoji: 🌖
-colorFrom: red
-colorTo: purple
-sdk: gradio
-sdk_version: 6.2.0
-app_file: app.py
-pinned: false
-license: mit
-short_description: Generate summaries in different styles
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+---
+title: Abstractive Style Summarizer
+emoji: ✍️
+colorFrom: blue
+colorTo: indigo
+sdk: gradio
+sdk_version: 6.2.0
+app_file: app.py
+pinned: false
+license: mit
+short_description: Generate summaries in different styles
+---
+
+# Abstractive Style Summarizer
+
+A multi-style abstractive text summarization tool powered by the `lityops/Abstractive-Style-Summarizer` model (fine-tuned Flan-T5 Base).
+
+## Features
+
+- **Harsh**: Very concise summaries, effectively extracting the core gist.
+- **Balanced**: A middle ground, balancing detail and brevity.
+- **Detailed**: Longer summaries preserving more context and information.
+
+## Usage
+
+Enter your text in the input box, select your desired style, and click "Process Summary".

app.py

@@ -1,104 +1,90 @@
-import gradio as gr
-from transformers import pipeline
-
-MODEL_ID = 'lityops/Style-Summarizer'
-
-summarizer = pipeline("summarization", model=MODEL_ID)
-
-def generate_summary(text, style):
-  if not text or len(text.strip()) < 50:
-    return "Input must at least be 50 words long"
-
-  input_text = f"summarize {style}: {text}"
-  input_words = len(text.split())
-
-  if style == 'harsh':
-    max_len = int(input_words * 0.35)
-    min_len = 5
-    rep_penalty = 2.5
-    length_penalty = 1.5
-    beam_size = 4
-    max_cap = 120
-  elif style == 'balanced':
-    max_len = int(input_words * 0.50)
-    min_len = 20
-    rep_penalty = 1.5
-    length_penalty = 1.2
-    beam_size = 4
-    max_cap = 180
-  else:
-    max_len = int(input_words * 0.70)
-    min_len = 50
-    rep_penalty = 1.2
-    length_penalty = 0.8
-    beam_size = 4
-    max_cap = 256
-
-  max_len = min(max_len, max_cap)
-  
-  output = summarizer(
-    input_text,
-    max_length=max_len, 
-    min_length=min_len, 
-    num_beams=beam_size,
-    length_penalty=length_penalty,
-    repetition_penalty=rep_penalty,
-    no_repeat_ngram_size=3,
-    early_stopping=True
-  )
-  return output[0]["summary_text"]
-
-custom_css = """
-#header {text-align: center; margin-bottom: 25px;}
-.gradio-container {max-width: 95% !important;}
-footer {display: none !important;}
-"""
-
-with gr.Blocks() as demo:
-  with gr.Column(elem_id="header"):
-    gr.Markdown("# Style Summarizer")
-    gr.Markdown("Fine-tuned Flan-T5 model for multi-style document summarization.")
-
-  with gr.Row():
-    with gr.Column(scale=1):
-      input_box = gr.Textbox(
-        label="Input Text", 
-        placeholder="Enter text to be summarized...", 
-        lines=15
-      )
-      style_radio = gr.Radio(
-        choices=["harsh", "standard", "detailed"], 
-        label="Summary Type", 
-        value="standard"
-      )
-      with gr.Row():
-        clear_btn = gr.Button("Clear Input")
-        submit_btn = gr.Button("Process Summary", variant="primary")
-      
-    with gr.Column(scale=1):
-      output_box = gr.Textbox(
-        label="Output Summary", 
-        lines=18, 
-        interactive=False
-      )
-      copy_btn = gr.Button("Copy to Clipboard")
-
-  submit_btn.click(
-    fn=generate_summary, 
-    inputs=[input_box, style_radio], 
-    outputs=output_box
-  )
-
-  clear_btn.click(
-    fn=lambda: "",
-    inputs=None,
-    outputs=input_box
-  )
-
-  copy_btn.click(
-    fn=None,
-    inputs=[output_box],
-    js="(v) => { navigator.clipboard.writeText(v); }"
-  )
-
-demo.launch(css=custom_css, theme=gr.themes.Base())
+import gradio as gr
+from transformers import pipeline
+
+MODEL_ID = 'lityops/Abstractive-Style-Summarizer'
+
+summarizer = pipeline("summarization", model=MODEL_ID)
+
+def generate_summary(text, style):
+  if not text or len(text.strip()) < 50:
+    return "Input must at least be 50 words long"
+
+  input_text = f"Summarize {style}: {text}"
+  input_words = len(text.split())
+
+  if style == 'Harsh':
+    max_len = int(input_words * 0.35)
+    min_len = 5
+    rep_penalty = 2.5
+    length_penalty = 1.5
+    beam_size = 4
+    max_cap = 120
+  elif style == 'Balanced':
+    max_len = int(input_words * 0.50)
+    min_len = 20
+    rep_penalty = 1.5
+    length_penalty = 1.2
+    beam_size = 4
+    max_cap = 180
+  else:
+    max_len = int(input_words * 0.70)
+    min_len = 50
+    rep_penalty = 1.2
+    length_penalty = 0.8
+    beam_size = 4
+    max_cap = 256
+
+  max_len = min(max_len, max_cap)
+  
+  output = summarizer(
+    input_text,
+    max_length=max_len, 
+    min_length=min_len, 
+    num_beams=beam_size,
+    length_penalty=length_penalty,
+    repetition_penalty=rep_penalty,
+    no_repeat_ngram_size=3,
+    early_stopping=True
+  )
+  return output[0]["summary_text"]
+
+custom_css = """
+#header {text-align: center; margin-bottom: 25px;}
+.gradio-container {max-width: 1000px !important;}
+footer {display: none !important;}
+"""
+
+with gr.Blocks(css=custom_css, theme=gr.themes.Base()) as demo:
+  with gr.Column(elem_id="header"):
+    gr.Markdown("# Style Summarizer")
+    gr.Markdown("Fine-tuned Flan-T5 model for multi-style document summarization.")
+
+  with gr.Row():
+    with gr.Column(scale=1):
+      input_box = gr.Textbox(
+        label="Input Text", 
+        placeholder="Enter text to be summarized...", 
+        lines=12
+      )
+      style_radio = gr.Radio(
+        choices=["Harsh", "Balanced", "Detailed"], 
+        label="Summary Type", 
+        value="Balanced"
+      )
+      submit_btn = gr.Button("Process Summary", variant="primary")
+      
+    with gr.Column(scale=1):
+      output_box = gr.Textbox(
+        label="Output Summary", 
+        lines=15, 
+        interactive=False,
+        show_copy_button=True
+      )
+
+  submit_btn.click(
+    fn=generate_summary, 
+    inputs=[input_box, style_radio], 
+    outputs=output_box
+  )
+
+demo.launch()

requirements.txt

@@ -1,3 +1,3 @@
-gradio
-torch
-transformers
+gradio
+torch>=2.5.1
+transformers>=4.36.0