Update app.py

app.py

@@ -1,12 +1,14 @@
 import streamlit as st
 from transformers import pipeline
 
+
 # Summarization
 def summarization(text):
-    text_model = pipeline("summarization", model="ainize/bart-base-cnn")
-    summary = text_model(text, max_length=100, temperature=1.0)[0]["generated_text"]
-    return summary
+    text_model = pipeline("text-generation", model="ainize/bart-base-cnn")
+    summary = text_model(text, max_length=100, temperature=1.0, return_full_text=False)[0]
+    return {"generated_text": summary["generated_text"]}
 
+    
 # Sentiment Classification
 def sentiment_classification(summary):
     sentiment_model = pipeline("text-classification", model="wxrrrrrrr/finetunde_sentiment_analysis")