Update app.py

app.py

@@ -1,9 +1,9 @@
 import streamlit as st
-from transformers import BartForConditionalGeneration, BartTokenizer
+from transformers import BartTokenizer, BartForSequenceClassification
 
 def summarize_text(data, max_length):
     model_name = "facebook/bart-large-cnn"
-    model = BartForConditionalGeneration.from_pretrained(model_name)
+    model = BartForSequenceClassification.from_pretrained(model_name)
     tokenizer = BartTokenizer.from_pretrained(model_name)
 
     inputs = tokenizer(data, max_length=max_length, return_tensors="pt", truncation=True)