kermitting soon

util/app.py

@@ -1,19 +1,13 @@
 # app.py
 import gradio as gr
 import xgboost as xgb
+from xgboost import DMatrix
 from huggingface_hub import hf_hub_download
 from app_training_df_getter import create_app_user_training_df
 import pandas as pd
-import numpy as np
-from collections import Counter
-import os
-from sklearn.model_selection import train_test_split, GridSearchCV
+from sklearn.model_selection import train_test_split
 from sklearn.preprocessing import LabelEncoder
-from sklearn.metrics import classification_report, precision_score, recall_score, f1_score
-from xgboost import XGBClassifier, plot_importance
-import xgboost as xgb
 from helper import * 
-import helper as helpfun
 
 
 # Define champion list for dropdowns
@@ -238,6 +232,11 @@ def predict_champion(player_opgg_url, *champions):
         )
         print("type(X_test): ", type(X_test), "\n")
 
+        # Convert Pandas DataFrame to DMatrix
+        X_test = DMatrix(X_test)
+
+        print("type(X_test) after converting to DMatrix: ", type(X_test), "\n")
+
         print("Starting model prediction... \n")
 
         prediction = model.predict(X_test)