Update app.py

app.py

@@ -1,11 +1,12 @@
 import gradio as gr
 import torch
-from transformers import AutoModelForSequenceClassification, AutoTokenizer
+# from transformers import AutoModelForSequenceClassification, AutoTokenizer
+from transformers import BertTokenizer, BertForSequenceClassification
 
 # Load the model and tokenizer from Hugging Face
 model_name = "crimson78/spam_classifier"  # Replace with your model's actual name
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSequenceClassification.from_pretrained(model_name)
+tokenizer = BertTokenizer.from_pretrained(model_name)
+model = BertForSequenceClassification.from_pretrained(model_name)
 
 # Define label mapping (Assuming 0 = HAM, 1 = SPAM)
 LABELS = {0: "HAM", 1: "SPAM"}