hernanhgm's picture
cambio descripcion
dfc88d1
raw
history blame
877 Bytes
import numpy as np
import gradio as gr
import joblib
def clasificador(value):
model = joblib.load("model_longpico.pkl")
scaler = joblib.load("scaler_longpico.pkl")
data = np.array(value).reshape(1, -1)
data_scaled = scaler.transform(data)
pred = model.predict(data_scaled)
label = ['hembra', 'macho']
sexo = label[pred[0]]
return sexo
title = "Clasificador del sexo de buitre negro"
description = """
Esta aplicación se ha creado para clasificar el sexo de las arpías completamente desarrolladas.
Para ello basta con medir la longitud del pico en milímetros.
A ver que nos conocemos, MILÍMETROS. Si se mide en centímetros no funciona, tienen que ser MILÍMETROS.
"""
demo = gr.Interface(
fn=clasificador,
inputs=["number"],
outputs="text",
title=title,
description=description
)
demo.launch()