wahaha commited on
Commit
df7ebc5
·
1 Parent(s): 67a4f7d
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -47,7 +47,8 @@ def parse_args() -> argparse.Namespace:
47
 
48
 
49
  def run(
50
- image
 
51
  ) -> tuple[PIL.Image.Image]:
52
 
53
  curPath = os.path.abspath(os.path.dirname(__file__))
@@ -66,7 +67,7 @@ def main():
66
  args = parse_args()
67
 
68
 
69
- func = functools.partial(run)
70
  func = functools.update_wrapper(func, run)
71
 
72
 
 
47
 
48
 
49
  def run(
50
+ image,
51
+ checkpoint_dir: str,
52
  ) -> tuple[PIL.Image.Image]:
53
 
54
  curPath = os.path.abspath(os.path.dirname(__file__))
 
67
  args = parse_args()
68
 
69
 
70
+ func = functools.partial(run, checkpoint_dir='')
71
  func = functools.update_wrapper(func, run)
72
 
73