Lauraayu commited on
Commit
5942de7
·
verified ·
1 Parent(s): 3b12bc1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -10,8 +10,8 @@ tokenizer_bb = AutoTokenizer.from_pretrained("Lauraayu/News_Classi_Model")
10
  model_bb = AutoModelForSequenceClassification.from_pretrained("Lauraayu/News_Classi_Model")
11
 
12
  # Streamlit application title
13
- st.title("News Article Summarizer and Classifier")
14
- st.write("Enter a news article text to get its summary and category.")
15
 
16
  # Text input for user to enter the news article text
17
  article = st.text_area("News Article", height=300)
@@ -41,7 +41,6 @@ if st.button("Classify"):
41
  label_mapping = model_bb.config.id2label
42
  predicted_label = label_mapping[predicted_label_id]
43
 
44
- # Display the summary and classification result
45
- st.write("Summary:", summary)
46
  st.write("Category:", predicted_label)
47
 
 
10
  model_bb = AutoModelForSequenceClassification.from_pretrained("Lauraayu/News_Classi_Model")
11
 
12
  # Streamlit application title
13
+ st.title("News Article Classifier")
14
+ st.write("Enter a news article text to get its category.")
15
 
16
  # Text input for user to enter the news article text
17
  article = st.text_area("News Article", height=300)
 
41
  label_mapping = model_bb.config.id2label
42
  predicted_label = label_mapping[predicted_label_id]
43
 
44
+ # Display the classification result
 
45
  st.write("Category:", predicted_label)
46