| import gradio as gr |
| import xgboost as xgb |
| import numpy as np |
| from huggingface_hub import hf_hub_download |
|
|
| |
| repo_id = "YDluffy/lottery_prediction" |
| model_filename = "lottery_xgboost_model.ubj" |
| model_path = hf_hub_download(repo_id=repo_id, filename=model_filename) |
|
|
| |
| 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 |
|
|
| |
| 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) |
|
|
