from transformers import pipeline
import gradio as gr

# Load model
classifier = pipeline("text-classification", model="Georg4000/Octa")

def predict(text):
    return classifier(text)

iface = gr.Interface(
    fn=predict,
    inputs=gr.Textbox(lines=2, placeholder="Enter text here..."),
    outputs="label",
    title="Octagon Text Classification",
    description="A demo for the Octagon model trained on IMDB reviews."
)

iface.launch()