app.py

import gradio as gr
from transformers import AutoModelForSequenceClassification, AutoTokenizer

model = AutoModelForSequenceClassification.from_pretrained("inkleaves/spam_detection_model")
tokenizer = AutoTokenizer.from_pretrained("inkleaves/spam_detection_model")

def predict(text):
    inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
    outputs = model(**inputs)
    prediction = outputs.logits.argmax(dim=-1).item()
    return "Spam" if prediction == 1 else "Not Spam"

interface = gr.Interface(fn=predict, inputs="Text", outputs="Text")
interface.launch(share=True)