Spaces:
Running
Running
Create app.py
Browse files
app.py
ADDED
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import os
|
2 |
+
from pathlib import Path
|
3 |
+
import time
|
4 |
+
from flask import Flask, render_template
|
5 |
+
from update_predictions import load_model, main_task
|
6 |
+
|
7 |
+
app = Flask(__name__)
|
8 |
+
Config = {
|
9 |
+
"REPO_PATH": Path(__file__).parent.resolve(),
|
10 |
+
"MODEL_PATH": os.path.join(os.environ.get("SPACE_CACHE_DIR", "."), "Kronos_model"),
|
11 |
+
# 其他配置保持不变...
|
12 |
+
}
|
13 |
+
|
14 |
+
# 加载模型(全局一次)
|
15 |
+
predictor = load_model()
|
16 |
+
|
17 |
+
@app.route("/")
|
18 |
+
def index():
|
19 |
+
# 每次访问时更新预测(或定时更新,见下文)
|
20 |
+
main_task(predictor)
|
21 |
+
return render_template("index.html")
|
22 |
+
|
23 |
+
if __name__ == "__main__":
|
24 |
+
app.run(host="0.0.0.0", port=7860)
|