File size: 448 Bytes
90331f6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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="your-username/octagon")

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