jemole's picture
Update app.py
b841471 verified
raw
history blame
736 Bytes
import gradio as gr
import pandas as pd
from joblib import load
def predict_bodymass(FlipperLength):
model = load("penguin_predictor.jb")
# Create DataFrame from input
data = {
"FlipperLength": [FlipperLength]
}
xin = pd.DataFrame(data)
bodymass = model.predict(xin)
return bodymass[0]
iface = gr.Interface(
fn=predict_bodymass,
inputs=[
gr.inputs.Textbox(placeholder="Introduce la longitud de la aleta en milímetros",numeric=True,label="Longitud de la aleta (mm)")
],
title="Modelo de regresión para el cálculo de masa corporal (gramos) de pingüinos",
outputs=["text"],
examples=[[180], [200], [220]]
)
if __name__ == "__main__":
iface.launch()