Lauraayu commited on
Commit
6dee087
·
verified ·
1 Parent(s): 8056289

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,8 +6,8 @@ import torch
6
  summarizer_ntg = pipeline("summarization", model="mrm8488/t5-base-finetuned-summarize-news")
7
 
8
  # Load the tokenizer and model for classification
9
- tokenizer_bb = AutoTokenizer.from_pretrained("your-username/your-model-name")
10
- model_bb = AutoModelForSequenceClassification.from_pretrained("your-username/your-model-name")
11
 
12
  # Streamlit application title
13
  st.title("News Article Summarizer and Classifier")
 
6
  summarizer_ntg = pipeline("summarization", model="mrm8488/t5-base-finetuned-summarize-news")
7
 
8
  # Load the tokenizer and model for classification
9
+ 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")