Update app.py

app.py

@@ -1,11 +1,11 @@
 import streamlit as st
-from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
 import torch
 
-# Load pre-trained DistilBERT model and tokenizer
-model_name = "sofzcc/distilbert-base-uncased-fake-news-checker"
-tokenizer = DistilBertTokenizer.from_pretrained(model_name)
-model = DistilBertForSequenceClassification.from_pretrained(model_name)
+# Load model directly
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+
+tokenizer = AutoTokenizer.from_pretrained("sofzcc/distilbert-base-uncased-fake-news-checker")
+model = AutoModelForSequenceClassification.from_pretrained("sofzcc/distilbert-base-uncased-fake-news-checker")
 
 # Function to predict if news is real or fake
 def predict_news(news_text):