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 águila arpía"
description = """
<img src="https://huggingface.co/spaces/hernanhgm/Sexaje_por_biometria/blob/main/imagen_arpia.jpg" alt="arpia" width="200"/>
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.
"""

demo = gr.Interface(
    fn=clasificador,
    inputs=["number"],
    outputs="text",
    title=title,
    description=description
)
demo.launch()