alexvaroz's picture
Update app.py
e1af253
raw
history blame
446 Bytes
from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
import gradio as grad
import ast
mdl_name = "lxyuan/distilbert-base-multilingual-cased-sentiments-student"
my_pipeline = pipeline(model=mdl_name, return_all_scores=True)
def get_text(text):
response = my_pipeline(text)
return response
grad.Interface(get_text, inputs=gr.Textbox(lines=2, placeholder="Name Here..."),
outputs="text").launch()