waritkan's picture
Update app.py
2d6c894 verified
import gradio as gr
import joblib
import pandas as pd
import numpy as np
# Load the model and unique brand values
model = joblib.load('model.joblib')
# Define the prediction function
def predict(p_yds, p_cmp,p_att,ints,cmp_pct,rate,p_adj_ypa,p_ypa,r_att,year_drafted) :
y_pds = int(p_yds)
p_cmp = int(p_cmp)
p_att = int(p_att)
ints = int(ints)
cmp_pct = float(cmp_pct)
rate = float(rate)
p_adj_ypa = float(p_adj_ypa)
p_ypa = float(p_ypa)
r_att = int(r_att)
year_drafted = int(year_drafted)
input_data = pd.DataFrame({
'p_yds': [p_yds],
'p_cmp': [p_cmp],
'p_att': [p_att],
'ints': [ints],
'cmp_pct': [cmp_pct],
'rate': [rate],
'p_adj_ypa': [p_adj_ypa],
'p_ypa': [p_ypa],
'r_att': [r_att],
'year_drafted': [year_drafted]
})
# Perform the prediction
prediction = model.predict(input_data)
return str(prediction[0])
# Create the Gradio interface
interface = gr.Interface(
fn=predict,
inputs=[
gr.Textbox(label="Pass Yards"),
gr.Textbox(label="Pass Completions"),
gr.Textbox(label="Pass Attempts"),
gr.Textbox(label="Interceptions"),
gr.Textbox(label="Completion Percentage"),
gr.Textbox(label="Passer Rating"),
gr.Textbox(label="Adjusted Yards per Attempt"),
gr.Textbox(label="Pass Yards per Attempt"),
gr.Textbox(label="Rush Attempts"),
gr.Textbox(label="Year Drafted")
],
outputs="textbox",
title="passing to touchdown Predictor",
description="Enter all informations."
)
# Launch the app
interface.launch()