intent_amazon / app.py
Bruno's picture
Update app.py
99596e7
raw
history blame contribute delete
693 Bytes
import gradio as gr
from transformers import pipeline
pipe = pipeline("text-classification", model="Bruno/intent-classification-pt-setfit-model")
dictionary={'48':'alarm_set',
'46':'audio_volume_mute',
'1':'iot_hue_lightchange',
'40':'iot_hue_lightoff',
'34':'iot_cleaning',
}
def greet(text, dictionary=dictionary):
response = pipe(text)
return dictionary[response[0]['label']]
demo = gr.Interface(fn=greet,
inputs="text",
outputs="text",
title="Aamzon PT-br Intent Classifier",
description="Aamzon PT-br Intent Classifier",
examples=[["pausa por dez segundos"],["definir a iluminação adequada para leitura"],["aspirar a casa"]],
cache_examples=False)
demo.launch()