Update app.py

app.py

@@ -5,8 +5,8 @@ from transformers import pipeline
 # Summarization
 def summarization(text):
     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"]}
+    summary = text_model(text, max_length=100, temperature=1.0)[0]["generated_text"]
+    return summary
 
     
 # Sentiment Classification