Update app.py

app.py

@@ -7,8 +7,8 @@ import torch
 tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-summarize-news")
 model = AutoModelWithLMHead.from_pretrained("mrm8488/t5-base-finetuned-summarize-news")
 
-tokenizer_bb = AutoTokenizer.from_pretrained("Lauraayu/News_Classi_Model")
-model_bb = AutoModelForSequenceClassification.from_pretrained("Lauraayu/News_Classi_Model")
+tokenizer_bb = AutoTokenizer.from_pretrained("Lauraayu/News_Classification_Model")
+model_bb = AutoModelForSequenceClassification.from_pretrained("Lauraayu/News_Classification_Model")
 
 # Streamlit application title
 st.title("News Article Classifier")
@@ -43,5 +43,6 @@ if st.button("Classify"):
     predicted_label = label_mapping[predicted_label_id]
     
     # Display the classification result
+    st.write("Summary:", summary)
     st.write("Category:", predicted_label)