File size: 441 Bytes
dfa0cd0 9c9b80a dfa0cd0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
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() |