fiewolf1000 commited on
Commit
17b3fee
·
verified ·
1 Parent(s): d6cba9b

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -0
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)