init

app.py

@@ -54,10 +54,10 @@ def run(
     checkpoint_dir = os.path.join(curPath, 'animeganv2/checkpoint/generator_Shinkai_weight')
     print(checkpoint_dir)
 
-    test.test(checkpoint_dir, 'save_dir', image.name, True)
+    outs = test.test(checkpoint_dir, 'save_dir', image.name, True)
 
 
-    return PIL.Image.open(image.name)
+    return PIL.Image.open(outs[0])
 
 
 def main():