mike10's picture
Update app.py
dbceef5
raw
history blame
No virus
362 Bytes
import gradio as gr
from transformers import pipeline
from presentation import main_title, examples
model = 'hackathon-pln-es/Detect-Acoso-Twitter-Es'
def classify(text):
cls= pipeline("text-classification", model=model)
return cls(text)[0]['label']
gr.Interface(fn=classify, inputs=["textbox"], outputs="text", examples=examples).launch(debug=True)