import gradio as gr
from transformers import pipeline

classifier = pipeline("text-classification", model="HooshvareLab/bert-fa-base-uncased-clf-persiannews")

def classify_text(text):
    result = classifier(text)
    # Return in a consistent format
    return result[0]["label"]  # Just returns the label as a string

iface = gr.Interface(fn=classify_text, inputs="text", outputs="text", title="Persian News Classifier")
iface.launch()