Update app.py

app.py

@@ -3,9 +3,9 @@ from transformers import DistilBertTokenizer, DistilBertForSequenceClassificatio
 import torch
 
 # Load pre-trained DistilBERT model and tokenizer
-model_name = "distilbert-base-uncased"
+model_name = "sofzcc/distilbert-base-uncased-fake-news-checker"
 tokenizer = DistilBertTokenizer.from_pretrained(model_name)
-model = DistilBertForSequenceClassification.from_pretrained(model_name, num_labels=2)
+model = DistilBertForSequenceClassification.from_pretrained(model_name)
 
 # Function to predict if news is real or fake
 def predict_news(news_text):
@@ -29,4 +29,3 @@ if st.button("Evaluate"):
         st.write(f"The news article is predicted to be: **{prediction}**")
     else:
         st.write("Please enter some news text to evaluate.")
-