game-predection / app.py
soni0391s's picture
Upload 3 files
fa3f863 verified
raw
history blame contribute delete
874 Bytes
import gradio as gr
import joblib
import numpy as np
# Load trained model and scaler
model = joblib.load("game_model.joblib")
scaler = joblib.load("scaler.joblib")
# Prediction function
def predict_sales(na, eu, jp, other, year):
X = np.array([[na, eu, jp, other, year]])
X_scaled = scaler.transform(X)
pred = model.predict(X_scaled)[0]
return float(pred)
# Gradio UI
interface = gr.Interface(
fn=predict_sales,
inputs=[
gr.Number(label="NA Sales"),
gr.Number(label="EU Sales"),
gr.Number(label="JP Sales"),
gr.Number(label="Other Sales"),
gr.Number(label="Year")
],
outputs=gr.Number(label="Predicted Global Sales"),
title="Game Sales Predictor (KNN Model)",
description="Enter sales values to predict the total global sales of a game."
)
interface.launch()