Update app.py

app.py

@@ -1,20 +1,30 @@
 import gradio as gr
 import joblib
+from huggingface_hub import hf_hub_download
+import numpy as np
 
-# 加载训练好的模型
-model = joblib.load("lottery_predictor_model.pkl")
+# 从 Hugging Face 下载模型
+model_path = hf_hub_download(repo_id="YDluffy/lottery_predictor_model", filename="lottery_predictor_model.pkl")
+model = joblib.load(model_path)
 
-# 定义预测函数
-def predict(lottery_period, date):
-    # 将期号和日期转化为模型需要的特征
-    features = [[lottery_period, date]]  # 转换为二维数组
-    prediction = model.predict(features)  # 使用模型进行预测
-    return f"Predicted numbers: {prediction}"
+# 预测函数
+def predict_lottery(year, period, num1, num2, num3, num4, num5, num6, special):
+    features = np.array([[year, period, num1, num2, num3, num4, num5, num6, special]])
+    prediction = model.predict(features)
+    return prediction
 
-# 创建 Gradio 接口
-interface = gr.Interface(fn=predict,
-                         inputs=[gr.Textbox(label="Lottery Period"), gr.Textbox(label="Date")],
-                         outputs="text")
+# Gradio Web 界面
+iface = gr.Interface(
+    fn=predict_lottery,
+    inputs=[
+        gr.Number(label="年份"), gr.Number(label="期数"),
+        gr.Number(label="号码1"), gr.Number(label="号码2"), gr.Number(label="号码3"),
+        gr.Number(label="号码4"), gr.Number(label="号码5"), gr.Number(label="号码6"),
+        gr.Number(label="特别号码")
+    ],
+    outputs="text",
+    title="六合彩预测模型",
+    description="输入期号和历史开奖号码，预测下一期开奖"
+)
 
-# 启动界面
-interface.launch()
+iface.launch()