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()