hylee commited on
Commit
7716d12
1 Parent(s): 22ce178
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -73,7 +73,7 @@ def main():
73
 
74
  args = parse_args()
75
 
76
- cartoonize = Cartoonize('file/wbc/saved_models/')
77
 
78
  func = functools.partial(run, cartoonize=cartoonize)
79
  func = functools.update_wrapper(func, run)
 
73
 
74
  args = parse_args()
75
 
76
+ cartoonize = Cartoonize(os.path.join(os.path.dirname(os.path.abspath(__file__)),'wbc/saved_models/'))
77
 
78
  func = functools.partial(run, cartoonize=cartoonize)
79
  func = functools.update_wrapper(func, run)