Update app.py

app.py

@@ -1,8 +1,15 @@
 from transformers import pipeline
 import gradio as gr
 
-# Load the TensorFlow-compatible pipeline
-pipe = pipeline("text-classification", model="ZachBeesley/Spam-Detector", framework="tf")
+# Use a pipeline as a high-level helper
+
+pipe = pipeline("text-classification", model="ZachBeesley/Spam-Detector")
+
+# Load model directly
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+
+tokenizer = AutoTokenizer.from_pretrained("ZachBeesley/Spam-Detector")
+model = AutoModelForSequenceClassification.from_pretrained("ZachBeesley/Spam-Detector")
 
 # Function to process the input text and return the predicted label
 def predict(text):