Update app.py

app.py

@@ -1,34 +1,29 @@
 import streamlit as st
 from transformers import pipeline
 
-# 1. Config
 MODEL_ID = "VoltIC/Automated-Text-Summarizer"
 
-# 2. Page Config (Must be first)
-st.set_page_config(page_title="AI Summarizer", layout="wide")
-
-# 3. Cached Loader
 @st.cache_resource
-def load_model():
-    # We use the subfolder since that's where your files are
-    return pipeline("summarization", model=MODEL_ID, subfolder="summarizer_model", framework="pt")
+def load_summarizer():
+    # Notice: 'subfolder' is GONE. This is much more stable.
+    return pipeline("summarization", model=MODEL_ID, framework="pt")
 
+st.set_page_config(page_title="AI Summarizer", page_icon="📝")
 st.title("📝 Automated Text Summarizer")
 
-# 4. Error-Resistant Loading
 try:
-    with st.spinner("Downloading AI Model (1.6GB)... This may take 3-5 minutes on first run."):
-        summarizer = load_model()
-    st.success("AI Model Loaded!")
+    # This will now find the files instantly at the root
+    summarizer = load_summarizer()
+    st.success("Model Loaded Successfully!")
 except Exception as e:
-    st.warning("The model is still downloading in the background. Please wait 2 minutes and refresh.")
+    st.error("The model is still initializing. Please wait 2 minutes.")
     st.stop()
 
-# 5. UI
-input_text = st.text_area("Paste text here:", height=250)
+input_text = st.text_area("Paste text here:", height=300)
+
 if st.button("Summarize"):
-    if input_text:
-        with st.spinner("Summarizing..."):
-            output = summarizer(input_text, max_length=100, min_length=30, do_sample=False)
-            st.subheader("Summary")
-            st.write(output[0]['summary_text'])
+    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'])