wahaha commited on
Commit
8357abe
1 Parent(s): d67e05e
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -48,7 +48,7 @@ def parse_args() -> argparse.Namespace:
48
  def run(
49
  image,
50
  ) -> tuple[PIL.Image.Image]:
51
-
52
  out = test.test(checkpoint_dir=os.path.join(curPath,'animeganv2/checkpoint/generator_Shinkai_weight'),
53
  style_name='Shinkai', test_file=image.name, if_adjust_brightness=True)
54
 
 
48
  def run(
49
  image,
50
  ) -> tuple[PIL.Image.Image]:
51
+ curPath = os.path.abspath(os.path.dirname(__file__))
52
  out = test.test(checkpoint_dir=os.path.join(curPath,'animeganv2/checkpoint/generator_Shinkai_weight'),
53
  style_name='Shinkai', test_file=image.name, if_adjust_brightness=True)
54