Update app.py

app.py

@@ -1,35 +1,18 @@
-import streamlit as st
-from transformers import pipeline
-
-st.set_page_config(page_title="AI TL;DR Tool", page_icon="📝")
-@st.cache_resource
-def load_summarizer():
-    return pipeline("summarization", model="./summarizer_model")
-
-summarizer = load_summarizer()
-
-st.title("Automated Text Summarizer")
-st.subheader("Generate a 2-3 sentence summary of long articles or legal docs.")
-
-text_to_summarize = st.text_area("Paste your long text here:", height=300)
-
-with st.sidebar:
-    st.header("Settings")
-    max_len = st.slider("Max Summary Length", 30, 150, 75)
-    min_len = st.slider("Min Summary Length", 10, 50, 25)
-
-if st.button("Generate TL;DR"):
-    if text_to_summarize.strip():
-        with st.spinner("Analyzing and condensing..."):
-            output = summarizer(
-                text_to_summarize, 
-                max_length=max_len, 
-                min_length=min_len, 
-                do_sample=False
-            )
-            
-            st.success("Summary Generated")
-            st.write(f"### Result:")
-            st.write(output[0]['summary_text'])
-    else:
-        st.warning("Please paste some text first!")
+import streamlit as st
+from transformers import pipeline
+
+MODEL_ID = "VoltIC/Automated-Text-Summarizer"
+
+@st.cache_resource
+def load_summarizer():
+    return pipeline("summarization", model=MODEL_ID)
+
+summarizer = load_summarizer()
+
+st.title("AI Text Summarizer")
+input_text = st.text_area("Paste text here...")
+
+if st.button("Summarize"):
+    if input_text:
+        summary = summarizer(input_text, max_length=100, min_length=30)
+        st.write(summary[0]['summary_text'])