Update app.py

app.py

@@ -1,30 +1,34 @@
 import os
-from huggingface_hub import hf_hub_download
+import requests
 
 def download_model():
-    # 使用绝对路径
-    model_dir = "/app/checkpoint"
-    model_path = os.path.join(model_dir, "final_model_K_30.pth")
+    model_path = "checkpoint/final_model_K_30.pth"
     
-    # 确保目录存在并设置权限
-    os.makedirs(model_dir, exist_ok=True)
-    os.chmod(model_dir, 0o777)
+    # 检查并修复目录权限问题
+    try:
+        os.makedirs("checkpoint", exist_ok=True)
+    except PermissionError:
+        print("无法创建目录，尝试使用当前目录")
+        model_path = "final_model_K_30.pth"  # 直接下载到当前目录
     
     if not os.path.exists(model_path):
-        print("使用 huggingface_hub 下载模型...")
+        print("下载模型文件...")
         try:
-            downloaded_path = hf_hub_download(
-                repo_id="Hanxiaofeng123/Deepcube",
-                filename="checkpoint/final_model_K_30.pth",
-                repo_type="space",
-                local_dir=model_dir,  # 下载到指定目录
-                local_dir_use_symlinks=False
-            )
-            print(f"模型下载完成: {downloaded_path}")
+            url = "https://huggingface.co/spaces/Hanxiaofeng123/Deepcube/resolve/main/checkpoint/final_model_K_30.pth"
+            
+            response = requests.get(url, stream=True, timeout=60)
+            response.raise_for_status()
+            
+            # 直接下载到文件
+            with open(model_path, 'wb') as f:
+                for data in response.iter_content(8192):
+                    f.write(data)
+            
+            print(f"下载完成: {model_path}")
             return True
             
         except Exception as e:
-            print(f"huggingface_hub 下载失败: {e}")
+            print(f"下载失败: {e}")
             return False
     else:
         print("模型文件已存在")