Spaces:

Lauraayu
/

NewsClassification_ISOM

Sleeping

Lauraayu commited on May 20, 2024

Commit

12208cf

verified ·

1 Parent(s): 1e03f32

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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_Classification_Model")
-model_bb = AutoModelForSequenceClassification.from_pretrained("Lauraayu/News_Classification_Model")
 # Streamlit application title
 st.title("News Article Classifier")

 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")
 # Streamlit application title
 st.title("News Article Classifier")