Fix pipeline task error: switch to summarization task

app.py

@@ -11,7 +11,7 @@ st.write("Summarize text, images, or PDFs with AI")
 
 @st.cache_resource
 def load_model():
-    return pipeline("text2text-generation", model="t5-small")
+    return pipeline("summarization", model="t5-small")
 
 summarizer = load_model()
 
@@ -33,9 +33,9 @@ def summarize_text(text):
     """Summarize the given text"""
     if not text.strip():
         return None
-    input_text = "summarize: " + text[:4000]
-    result = summarizer(input_text, max_length=150, min_length=40, do_sample=False)
-    return result[0]["generated_text"]
+    # input_text = "summarize: " + text[:4000]
+    result = summarizer(text[:4000], max_length=150, min_length=40, do_sample=False)
+    return result[0]["summary_text"]
 
 tab1, tab2, tab3 = st.tabs(["📝 Text", "🖼️ Image", "📄 PDF"])