init

app.py

@@ -49,12 +49,12 @@ def parse_args() -> argparse.Namespace:
 
 def run(
     image,
-    p2c : Photo2Cartoon,
+    p2c,
 ) -> tuple[PIL.Image.Image]:
 
     cartoon = p2c.inference(image.name)
 
-    return PIL.Image.fromarray(cartoon)
+    return PIL.Image.Image.fromarray(cartoon)
 
 
 def main():
@@ -64,7 +64,7 @@ def main():
 
     p2c = Photo2Cartoon()
 
-    func = functools.partial(run, p2c)
+    func = functools.partial(run, p2c=p2c)
     func = functools.update_wrapper(func, run)