| 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() |