Debug

app.py

@@ -25,11 +25,31 @@ model_repo = "kairunwen/LSM"  # Replace with the actual repository name
 model_filename = "checkpoint-40.pth"  # Model filename
 
 # Download model from HuggingFace
-model_path = hf_hub_download(repo_id=model_repo, filename=model_filename)
+# model_path = hf_hub_download(repo_id=model_repo, filename=model_filename)
 
 # Load model
-model = LSM_MASt3R.from_pretrained(model_path)
-model = model.eval()
+# model = LSM_MASt3R.from_pretrained(model_path)
+# model = model.eval()
+
+
+try:
+    # 下载模型文件
+    model_path = hf_hub_download(repo_id=model_repo, filename=model_filename)
+    print(f"模型文件已下载到: {model_path}")
+
+    # 加载模型
+    model = LSM_MASt3R.from_pretrained(model_path, device='cuda')
+    model = model.eval()
+    print("模型加载成功并设置为评估模式！")
+
+except FileNotFoundError:
+    print(f"错误: 无法找到文件 {model_filename}，请检查仓库 {model_repo} 是否正确上传文件。")
+except KeyError as e:
+    print(f"错误: 检查点文件格式不正确，缺少键 {e}。请确认 checkpoint-40.pth 包含 'args' 和 'model'。")
+except Exception as e:
+    print(f"发生未知错误: {e}")
+
+
 
 
 @spaces.GPU(duration=80)