Restart para arpía. Solo long pico

app.py

@@ -1,48 +1,34 @@
-import numpy as np
 import gradio as gr
-import joblib
 
 
-def promediado(L_izda, DV_izda, L_dcha, DV_dcha):
-    L = (L_izda+L_dcha)/2
-    DV = (DV_izda+DV_dcha)/2
-    return L, DV
-
-def clasificador(L, DV):
+def clasificador(measured_value):
     
-    model = joblib.load("model.pkl") 
-    scaler = joblib.load("scaler.pkl")
+    import numpy as np
+    from joblib import load
+ 
+ 
+    scaler = load("scaler_longpico.pkl")
+    model = load("model_longpico.pkl") 
     
-    data = np.array([L, DV]).reshape(1, -1)
+    data = np.array([measured_value]).reshape(1, -1)
     data_scaled = scaler.transform(data)
     pred = model.predict(data_scaled)
     
-    sexo = pred[0]
-    
-    return sexo
- 
-def clasificador_completo(L_izda, DV_izda, L_dcha, DV_dcha): 
-    L, DV = promediado(L_izda, DV_izda, L_dcha, DV_dcha)
-    sexo = clasificador(L, DV)
+    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 los pollos de buitres negros. 
-Para ello basta con tomar las medidas lateral y dorso-ventral de ambos tarsos del individuo.
-Los datos a introducir son los siguientes \n
-    L_izda: medida lateral del tarso de la pata izquierda \n
-    DV_izda: medida dorso-ventral del tarso de la pata izquierda \n
-    L_dcha: medida lateral del tarso de la pata derecha \n
-    DV_izda: medida dorso-ventral del tarso de la pata derecha
+Esta aplicación se ha creado para clasificar el sexo de las arpías completamente desarrolladas. 
+Para ello basta con medir la longitu del pico.
 """
 
 demo = gr.Interface(
-    fn=clasificador_completo,
-    inputs=["number", "number", "number", "number"],
+    fn=clasificador,
+    inputs=["number"],
     outputs="text",
     title=title,
     description=description
-)
-demo.launch()
+)