jefercania/image-classification-tensorflow

1import gradio as gr2import tensorflow as tf3import requests4from PIL import Image5import numpy as np6 7# Cargando el modelo8inception_net = tf.keras.applications.MobileNetV2()9 10# Obteniendo las etiquetas11respuesta = requests.get("https://git.io/JJkYN")12etiquetas = respuesta.text.split("\n")13 14def redimensionar_imagen(img_array, target_size=(224, 224)):15    img = Image.fromarray(img_array)16    img = img.resize(target_size)17    return np.array(img)18 19def clasifica_imagen(inp):20    # Redimensionar la imagen21    inp = redimensionar_imagen(inp)22    23    # Verificar la forma actual de la imagen24    if inp.shape != (224, 224, 3):25        raise ValueError(f"Expected input shape (224, 224, 3), but got {inp.shape}")26    27    # Hacer prediccion28    inp = tf.keras.applications.mobilenet_v2.preprocess_input(inp)29    prediction = inception_net.predict(inp.reshape((-1, 224, 224, 3))).flatten()30    confidences = {etiquetas[i]: float(prediction[i]) for i in range(1000)}31    return confidences32 33demo = gr.Interface(fn=clasifica_imagen,34                    inputs=gr.Image(),35                    outputs=gr.Label(num_top_classes=3),36                   )37 38demo.launch(debug=True)39