Update app.py

app.py

@@ -1,29 +1,33 @@
 import streamlit as st
 from transformers import pipeline
 
+# Your verified Model ID
 MODEL_ID = "VoltIC/Automated-Text-Summarizer"
 
 @st.cache_resource
 def load_summarizer():
-    # Notice: 'subfolder' is GONE. This is much more stable.
-    return pipeline("summarization", model=MODEL_ID, framework="pt")
+    # This worked in your logs!
+    return pipeline(
+        "summarization", 
+        model=MODEL_ID, 
+        subfolder="summarizer_model", 
+        framework="pt"
+    )
 
-st.set_page_config(page_title="AI Summarizer", page_icon="📝")
+st.set_page_config(page_title="AI TL;DR", page_icon="📝")
 st.title("📝 Automated Text Summarizer")
 
-try:
-    # This will now find the files instantly at the root
-    summarizer = load_summarizer()
-    st.success("Model Loaded Successfully!")
-except Exception as e:
-    st.error("The model is still initializing. Please wait 2 minutes.")
-    st.stop()
+# Load the model (The logs prove this is working now!)
+summarizer = load_summarizer()
 
-input_text = st.text_area("Paste text here:", height=300)
+text_input = st.text_area("Paste long text here:", height=300)
 
 if st.button("Summarize"):
-    if input_text.strip():
-        with st.spinner("Processing..."):
-            summary = summarizer(input_text, max_length=100, min_length=30)
-            st.write("### Summary:")
-            st.write(summary[0]['summary_text'])
+    if text_input.strip():
+        with st.spinner("AI is thinking..."):
+            # length_penalty=2.0 helps keep it to 2-3 sentences
+            summary = summarizer(text_input, max_length=100, min_length=30, length_penalty=2.0)
+            st.success("Summary Generated!")
+            st.markdown(f"### Result:\n{summary[0]['summary_text']}")
+    else:
+        st.warning("Please enter text first.")