app.py

import torch
import gradio as gr
from transformers import pipeline

# Load moderation model
moderator = pipeline("text-classification", model="KoalaAI/Text-Moderation")

# Define function
def moderate_text(input_text):
    result = moderator(input_text)
    label = result[0]['label']
    score = round(result[0]['score'] * 100, 2)
    return f"Label: {label}, Confidence: {score}%"

# Gradio interface
demo = gr.Interface(fn=moderate_text, inputs="text", outputs="text", title="AISA - Text Moderation")
demo.launch()