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()