Update app.py

app.py

@@ -1,11 +1,13 @@
 import gradio as gr
 import joblib
 from huggingface_hub import hf_hub_download
+import xgboost as xgb
 import numpy as np
 
-# 从 Hugging Face 下载模型
-model_path = hf_hub_download(repo_id="YDluffy/lottery_predictor_model", filename="lottery_predictor_model.pkl")
-model = joblib.load(model_path)
+# 下载并加载模型
+model_path = hf_hub_download(repo_id="YDluffy/lottery_prediction", filename="lottery_xgboost_model.json")
+model = xgb.XGBRegressor()
+model.load_model(model_path)
 
 # 预测函数
 def predict_lottery(year, period, num1, num2, num3, num4, num5, num6, special):
@@ -13,7 +15,7 @@ def predict_lottery(year, period, num1, num2, num3, num4, num5, num6, special):
     prediction = model.predict(features)
     return prediction
 
-# Gradio Web 界面
+# 创建 Gradio 界面
 iface = gr.Interface(
     fn=predict_lottery,
     inputs=[