Spaces:
Sleeping
Sleeping
from huggingface_hub import from_pretrained_fastai | |
import gradio as gr | |
from fastai.vision.all import * | |
# repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME" | |
repo_id = "allopeap/emociones" | |
learner = from_pretrained_fastai(repo_id) | |
labels = learner.dls.vocab | |
# Definimos una función que se encarga de llevar a cabo las predicciones | |
def predict(text): | |
pred,pred_idx, probs = learner.predict(text) | |
if pred == '0': | |
return "Positive" | |
elif pred == 1: | |
return "Neutral" | |
else: | |
return "Negative" | |
examp = ["Todo el mundo con la segunda temporada de Narcos y yo no, PORQUE NO TENGO NETFLIX @user pasaros una suscripcion de 5 dias", | |
"Me gusta estar sola y me molesta la gente que habla mucho. Si tan solo nos juntáramos a escuchar música y de rato en rato hablar de ello", | |
"La voz sepsi que debe tener Que dicha Gaby! Ya el lunes regresará con toda la pata"] | |
# Creamos la interfaz y la lanzamos. | |
gr.Interface( | |
fn=predict, | |
inputs=gr.inputs.Textbox(lines=1, placeholder="Ingresa el texto aquí..."), | |
outputs=gr.Textbox(label="Prediction"), | |
examples=examp | |
).launch(share=False) | |