Spaces:
Sleeping
Sleeping
import gradio as gr | |
from transformers import pipeline | |
"""pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es") | |
def predict(text): | |
return pipe(text)[0]["translation_text"] | |
demo = gr.Interface( | |
fn=predict, | |
inputs='text', | |
outputs='text', | |
)""" | |
pipe = pipeline("sentiment-analysis", model="michellejieli/emotion_text_classifier") | |
def predict(text): | |
return pipe(text)[0]["label"], pipe(text)[0]['score'] | |
demo = gr.Interface( | |
fn = predict, | |
inputs = gr.Textbox(placeholder="Text here ..."), | |
outputs = ['text', 'number'] | |
) | |
demo.launch() | |