Update app.py

app.py

@@ -58,15 +58,19 @@ def teed_process_image(image):
     args.input_val_dir = './teed_tmp'
     args.output_dir = './output/teed_imgs'
 
-    print(args)  # 调试信息
-    print(train_info)  # 调试信息
+    checkpoint_path = './output/teed_imgs/BIPED/5/5_model.pth'
+    if not os.path.exists(checkpoint_path):
+        raise FileNotFoundError(f"Checkpoint file not found: {checkpoint_path}")
+    
+    args.checkpoint_data = checkpoint_path  # 确保使用正确的路径
 
-    main(args, train_info)  # 确保调用正确
+    main(args, train_info)
 
     shutil.rmtree('./teed_tmp')
     return cv2.imread(os.path.join('./output/teed_imgs', 'processed_image.png'))
 
 
+
 # 处理单个图像
 def process_single_image(image):
     depth_result = depth_anything_image(image, 'vitl')