julenalvaro's picture
Agregada función de clasificar imágenes (#1)
1ebca6a
import gradio as gr
from transformers import pipeline
from funciones import clasifica_imagen, audio_a_text, texto_a_sentimiento
demo = gr.Blocks()
with demo:
gr.Markdown('Mini app hecha con Gradio. Contiene tres pestañas en las que se puede: 1 ) transcribir texto en castellano 2) Calificar el sentimiento de una frase entre negativo, neutro y positivo y 3) Comprobar si una imagen es un perro o un gato.')
with gr.Tabs(): #vamos a crear pestañas
with gr.TabItem('Transcribe audio en español'):
with gr.Row():
audio = gr.Audio(source = 'microphone', type = 'filepath')
transcription = gr.Textbox()
b1 = gr.Button('Transcribe audio')
with gr.TabItem('Análisis del sentimiento en español'):
with gr.Row():
texto = gr.Textbox()
label = gr.Label()
b2 = gr.Button('Dime el sentimiento por favor')
with gr.TabItem('Clasificador de imágenes perros VS gatos'):
with gr.Row():
image = gr.Image(shape = (224,224))
lb_image = gr.Label(num_top_classes = 3)
b3 = gr.Button("Clasifica un perro o un gato")
b1.click(audio_a_text, inputs = audio, outputs = transcription)
b2.click(texto_a_sentimiento, inputs = texto, outputs = label)
b3.click(clasifica_imagen, inputs = image, outputs = lb_image)
demo.launch()