File size: 1,108 Bytes
72b3a96 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
import gradio as gr
import xgboost as xgb
import numpy as np
from huggingface_hub import hf_hub_download
# **📥 从 Hugging Face 下载 XGBoost 模型**
repo_id = "YDluffy/lottery_prediction"
model_filename = "lottery_xgboost_model.ubj"
model_path = hf_hub_download(repo_id=repo_id, filename=model_filename)
# **✅ 加载 XGBoost 模型**
model = xgb.Booster()
model.load_model(model_path)
# **📌 预测函数**
def predict_lottery(year, period, num1, num2, num3, num4, num5, num6, special):
test_features = np.array([[year, period, 1, 1, num1, num2, num3, num4, num5, num6, special, 30]])
dtest = xgb.DMatrix(test_features)
prediction = model.predict(dtest)
final_prediction = np.round(prediction).astype(int).tolist()
return final_prediction
# **📌 创建 API 接口**
iface = gr.Interface(
fn=predict_lottery,
inputs=["number", "number", "number", "number", "number", "number", "number", "number", "number"],
outputs="text",
title="六合彩预测 API",
description="输入年份、期号和开奖号码，返回预测结果"
)
iface.launch(share=True)
|