Lauraayu commited on
Commit
9a28330
·
verified ·
1 Parent(s): e1fb94e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -3,11 +3,11 @@ from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassifica
3
  import torch
4
 
5
  # Define the summarization pipeline
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")
 
3
  import torch
4
 
5
  # Define the summarization pipeline
6
+ summarizer_ntg = pipeline("text2text-generation", 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")