| import gradio as gr |
| import tensorflow as tf |
| from huggingface_hub import from_pretrained_keras |
| import numpy as np |
|
|
| model = from_pretrained_keras("keras-io/semi-supervised-classification-simclr") |
|
|
| labels_gradio = ["Avi贸n", "Pajaro", "Coche", "Gato", "Ciervo", "Perro", "Caballo", "Mono", "Barco", "Cami贸n"] |
|
|
|
|
| def predict(imput_image): |
| image = tf.constant(imput_image) |
| image = tf.reshape(image, [-1, 96, 96, 3]) |
| pred = model.predict(image) |
| pred_list = pred[0, :] |
| pred_softmax = np.exp(pred_list)/np.sum(np.exp(pred_list)) |
| softmax_list = pred_softmax.tolist() |
| return {labels_gradio[i]: softmax_list[i] for i in range(10)} |
|
|
|
|
| image = gr.inputs.Image(shape=(96, 96)) |
| label = gr.outputs.Label(num_top_classes=4) |
|
|
| pie_pag = """<center> |
| Modelo: <a href='https://huggingface.co/keras-io/semi-supervised-classification-simclr' target='_blank'>keras.io</a> |
| Basado en el Space: <a href='https://huggingface.co/spaces/keras-io/semi-supervised-classification' target='_blank'>Andr谩s B茅res</a> |
| Autor: <a href='https://huggingface.co/machde' target='_blank'>Manuel Chac贸n De Dios</a>""" |
|
|
| titulo = "Mini clasificador" |
| descripcion = """<center>Clasificador capaz de etiquetar si es un Avi贸n, Pajaro, Coche, |
| Gato, Ciervo, Perro, Caballo, Mono, Barco, Cami贸n</center>""" |
|
|
|
|
|
|
|
|
| Iface = gr.Interface( |
| fn=predict, |
| inputs=image, |
| outputs=label, |
| layout="vertical", |
| theme="seafoam", |
| examples=[["test_img/pajaro-test.jpeg"], ["test_img/coche-test.jpg"], ["test_img/perro-test.jpg"]], |
| title=titulo, |
| article=pie_pag, |
| description=descripcion, |
| ).launch() |
|
|
