Lauraayu commited on
Commit
3b12bc1
·
verified ·
1 Parent(s): 5ced387

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
  tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-summarize-news")
7
  model = AutoModelWithLMHead.from_pretrained("mrm8488/t5-base-finetuned-summarize-news")
8
 
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
  tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-summarize-news")
7
  model = AutoModelWithLMHead.from_pretrained("mrm8488/t5-base-finetuned-summarize-news")
8
 
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")