Spaces:

KumaPower
/

AvatarArtist

Running on Zero

刘虹雨 commited on Apr 1

Commit

66886ce

1 Parent(s): 7492542

update code

Files changed (1) hide show

app.py CHANGED Viewed

@@ -654,7 +654,7 @@ def process_image(input_image_dir, source_type, is_style, save_dir):
         # 使用 difflib 查找相似文件名
         matches = difflib.get_close_matches(image_name_true, image_files, n=1, cutoff=0.1)
         closest_match = matches[0]
-        imge_dir = os.path.join(process_img_save_dir, 'processed_img/dataset/images512x512/input_image', closest_match)
         image = Image.open(imge_dir)
         image_name_true = closest_match
         return image, source_type, image_name_true  # 这里替换成 处理用户上传图片的逻辑

         # 使用 difflib 查找相似文件名
         matches = difflib.get_close_matches(image_name_true, image_files, n=1, cutoff=0.1)
         closest_match = matches[0]
+        imge_dir = os.path.join(process_img_save_dir, 'dataset/images512x512/input_image', closest_match)
         image = Image.open(imge_dir)
         image_name_true = closest_match
         return image, source_type, image_name_true  # 这里替换成 处理用户上传图片的逻辑