Update app.py

app.py

@@ -1,3 +1,28 @@
+import os
+from huggingface_hub import hf_hub_download
+
+def download_model():
+    model_path = "checkpoint/final_model_K_30.pth"
+    if not os.path.exists(model_path):
+        print("从Hugging Face Hub下载模型...")
+        try:
+            # 使用huggingface_hub下载
+            downloaded_path = hf_hub_download(
+                repo_id="Hanxiaofeng123/Deepcube",
+                filename="checkpoint/final_model_K_30.pth",
+                local_dir=".",
+                local_dir_use_symlinks=False
+            )
+            print(f"模型下载完成: {downloaded_path}")
+        except Exception as e:
+            print(f"下载失败: {e}")
+            # 尝试创建空文件继续运行（仅用于测试）
+            os.makedirs("checkpoint", exist_ok=True)
+            open(model_path, 'w').close()
+            print("创建了空的模型文件用于测试")
+
+download_model()
+
 import flask
 from flask import request, jsonify
 import torch