jhj0517 commited on
Commit
a79cb95
1 Parent(s): 63f92e8
Files changed (1) hide show
  1. tests/test_image_restoration.py +3 -2
tests/test_image_restoration.py CHANGED
@@ -14,7 +14,7 @@ from modules.live_portrait.live_portrait_inferencer import LivePortraitInference
14
  def test_image_restoration(
15
  input_image: str,
16
  ):
17
- if not os.path.exists(TEST_IMAGE_PATH):
18
  download_image(
19
  TEST_IMAGE_URL,
20
  TEST_IMAGE_PATH
@@ -23,7 +23,8 @@ def test_image_restoration(
23
  inferencer = LivePortraitInferencer()
24
 
25
  restored_output = inferencer.resrgan_inferencer.restore_image(
26
- input_image
 
27
  )
28
 
29
  assert os.path.exists(restored_output)
 
14
  def test_image_restoration(
15
  input_image: str,
16
  ):
17
+ if not os.path.exists(input_image):
18
  download_image(
19
  TEST_IMAGE_URL,
20
  TEST_IMAGE_PATH
 
23
  inferencer = LivePortraitInferencer()
24
 
25
  restored_output = inferencer.resrgan_inferencer.restore_image(
26
+ input_image,
27
+ overwrite=False
28
  )
29
 
30
  assert os.path.exists(restored_output)