text-emotion / app.py
macapa's picture
Update app.py
e186aa3
raw
history blame
661 Bytes
from fastai.text.all import *
from huggingface_hub import from_pretrained_fastai
import gradio as gr
# Cargamos el learner
repo_id = "macapa/emotion-text-classification"
learner = from_pretrained_fastai(repo_id)
# Definimos la funciΓ³n que realiza las predicciones
def predict(text):
pred = learner.predict(text)
return pred
texto1 = 'i can go from feeling so hopeless to so damned hopeful just from being around someone who cares and is awake'
texto2 = 'im grabbing a minute to post i feel greedy wrong'
# Creamos la interfaz y la lanzamos.
gr.Interface(fn=predict, inputs="text", outputs="text", examples=[texto1,texto2]).launch(share=False)