Spaces:

hylee
/

White-box-Cartoonization

Runtime error

hylee commited on Apr 29, 2022

Commit

7716d12

•

1 Parent(s): 22ce178

init

Files changed (1) hide show

app.py CHANGED Viewed

@@ -73,7 +73,7 @@ def main():
     args = parse_args()
-    cartoonize = Cartoonize('file/wbc/saved_models/')
     func = functools.partial(run, cartoonize=cartoonize)
     func = functools.update_wrapper(func, run)

     args = parse_args()
+    cartoonize = Cartoonize(os.path.join(os.path.dirname(os.path.abspath(__file__)),'wbc/saved_models/'))
     func = functools.partial(run, cartoonize=cartoonize)
     func = functools.update_wrapper(func, run)