Update app.py

app.py

@@ -237,24 +237,18 @@ def teed_imgs(img_path='./input', outdir='./output/teed_imgs', gaussianBlur=[0,
     os.makedirs(outdir, exist_ok=True)
     os.makedirs('teed_tmp', exist_ok=True)
 
-    # 处理上传的单个图片
+    # 处理单个上传的图片
     if os.path.isfile(img_path):
-        filenames = [img_path]
-    else:
-        filenames = os.listdir(img_path)
-        filenames = [os.path.join(img_path, filename) for filename in filenames if not filename.startswith('.') and filename.lower().endswith(('.jpg', '.jpeg', '.png', '.gif', '.bmp', '.tiff', '.webp', 'tif'))]
-        filenames.sort()
-
-    for filename in tqdm(filenames):
+        img = cv2.imread(img_path)
         if gaussianBlur[0] != 0:
-            img = cv2.imread(filename)
             img = cv2.GaussianBlur(img, (gaussianBlur[1], gaussianBlur[1]), gaussianBlur[2])
-            cv2.imwrite(os.path.join('teed_tmp', os.path.basename(filename)), img)
+            cv2.imwrite(os.path.join('teed_tmp', 'temp_image.png'), img)
         else:
-            shutil.copy(filename, 'teed_tmp')
+            cv2.imwrite(os.path.join('teed_tmp', 'temp_image.png'), img)
 
-    # 这里调用您的其他处理代码
+    # 使用临时图像文件进行处理
     args, train_info = parse_args(is_testing=True, pl_opt_dir=outdir)
+    args.input_val_dir = 'teed_tmp'  # 使用临时目录
     teed.main(args, train_info)
 
     shutil.rmtree('teed_tmp')