刘虹雨 commited on
Commit
7fa0d80
·
1 Parent(s): 5ccc87f

update code

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -639,8 +639,8 @@ def process_image(input_image_dir, source_type, is_style, save_dir):
639
  os.makedirs(process_img_save_dir, exist_ok=True)
640
  os.makedirs(process_img_input_dir, exist_ok=True)
641
  if source_type == "example":
642
- input_image = Image.open(input_image)
643
- return input_image, source_type, image_name_true
644
  else:
645
  # input_process_model.inference(input_image, process_img_save_dir)
646
  shutil.copy(input_image_dir, process_img_input_dir)
 
639
  os.makedirs(process_img_save_dir, exist_ok=True)
640
  os.makedirs(process_img_input_dir, exist_ok=True)
641
  if source_type == "example":
642
+ image = Image.open(input_image_dir)
643
+ return image, source_type, image_name_true
644
  else:
645
  # input_process_model.inference(input_image, process_img_save_dir)
646
  shutil.copy(input_image_dir, process_img_input_dir)