Spaces:
Sleeping
Sleeping
File size: 1,028 Bytes
6cfe2a9 f3ff304 6cfe2a9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | # Instale o Gradio: !pip install gradio
import gradio as gr
from transformers import pipeline
# Carrega o modelo (exatamente o mesmo código de antes)
classifier = pipeline("zero-shot-classification")
# Define a função que a interface web vai chamar
def classificar_emoji(frase):
labels = ["😄", "😭", "😠", "🤔", "🎉", "❤️", "🚀"]
resultado = classifier(frase, candidate_labels=labels)
# Formata o resultado para o Gradio
# (retorna um dicionário de {label: score})
return {label: score for label, score in zip(resultado['labels'], resultado['scores'])}
# Cria a interface
# Inputs: "text" (caixa de texto)
# Outputs: "label" (mostra os scores)
# Título e descrição
iface = gr.Interface(fn=classificar_emoji,
inputs="text",
outputs="label",
title="Classificador de Emojis 🤖",
description="Escreva uma frase e veja a IA sugerir o emoji perfeito.")
# Lança a aplicação web
iface.launch() |