IbraahimLab's picture
adding pandas
ca15d0d verified
import gradio as gr
import joblib
import pandas as pd
model=joblib.load("calorie_prediction_model.pkl")
def predict_calories(age, gender, height, weight, duration, heart_rate, body_temp):
data = pd.DataFrame([{
"Age": age,
"Gender": gender,
"Height": height,
"Weight": weight,
"Duration": duration,
"Heart_Rate": heart_rate,
"Body_Temp": body_temp
}])
prediction = model.predict(data)
return f"🔥 Estimated Calories Burned: {round(prediction[0], 2)}"
app = gr.Interface(
fn=predict_calories,
inputs=[
gr.Slider(10, 80, step=1, label="Age"),
gr.Radio(["male", "female"], label="Gender"),
gr.Slider(100, 220, step=1, label="Height (cm)"),
gr.Slider(30, 150, step=1, label="Weight (kg)"),
gr.Slider(1, 120, step=1, label="Exercise Duration (minutes)"),
gr.Slider(60, 200, step=1, label="Heart Rate (bpm)"),
gr.Slider(35.0, 42.0, step=0.1, label="Body Temperature (°C)")
],
outputs=gr.Textbox(label="🔥 Calories Prediction"),
title="🧠 Calorie Burn Predictor",
description="Fill in your workout data to predict how many calories you might burn."
)
app.launch()