Update app.py

app.py

@@ -1,3 +1,13 @@
+import torch
+from huggingface_hub import hf_hub_download
+
+# 下载模型文件（会自动缓存到 /root/.cache/huggingface/hub/）
+model_path = hf_hub_download(
+    repo_id="Hanxiaofeng123/Deepcube",  # 替换成你自己的仓库
+    repo_type="spaces",
+    filename="checkpoint/final_model_K_30.pth"
+)
+
 import flask
 from flask import request, jsonify
 import torch
@@ -13,12 +23,6 @@ app = flask.Flask(__name__, static_folder=None)
 app.config['JSON_AS_ASCII'] = False
 app.config['DEBUG'] = True
 
-# 加载配置
-config = Config()
-args = config.parse_args()
-# 设置默认模型路径
-model_path = args.model_path
-
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # 加载模型和创建Cube对象