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")