Update app.py

app.py

@@ -54,19 +54,18 @@ def teed_process_image(image):
     temp_image_path = './teed_tmp/temp_image.png'
     cv2.imwrite(temp_image_path, np.array(image))
 
-    args = parse_args(is_testing=True, pl_opt_dir='./output/teed_imgs')
+    # 获取解析后的参数
+    args, train_info = parse_args(is_testing=True, pl_opt_dir='./output/teed_imgs')
     args.input_val_dir = './teed_tmp'  # 临时目录
     args.output_dir = './output/teed_imgs'  # 输出目录
 
-    # 确保 TEED 处理的代码正常
-    if hasattr(teed, 'main'):
-        teed.main(args)
-    else:
-        print("TEED module does not have a main function.")
+    # 调用 TEED 主函数进行处理
+    teed.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')