Crie interfaces para seus modelos de ML com Gradio e HuggingFace

Com algumas linhas de código, construí uma interface simples e amigável para testar um modelo pré-treinado da plataforma huggingface usando Gradio. O resultado final é este:

É um modelo genérico que classifica um dado texto como positivo ou negativo (com melhor precisão em inglês); é possível escolher um com mais parâmetros usando o módulo transformers.

A interface foi feita com Gradio, uma ferramenta simples e intuitiva que permite criar interfaces para a prototipagem dos seus modelos de Machine Learning.

Para os aventureiros, disponibilizei meu código, e o desafio é o seguinte: ajuste a interface para outra entrada de dados (imagens), selecione um modelo específico de todos os disponíveis no Hugging Face e, por exemplo, faça a classificação das mesmas; ou crie uma interface que receba um texto e o transforme em imagens. O céu é o limite!

Código no huggingface: https://huggingface.co/spaces/CASLL/Sentiment-Analysis/blob/main/app.py