init

app.py

@@ -15,7 +15,9 @@ import numpy as np
 import PIL.Image
 
 from io import BytesIO
+sys.path.insert(0, 'animeganv2')
 
+import test as test
 
 ORIGINAL_REPO_URL = 'https://github.com/TachibanaYoshino/AnimeGANv2'
 TITLE = 'TachibanaYoshino/AnimeGANv2'
@@ -48,6 +50,12 @@ def run(
     image,
 ) -> tuple[PIL.Image.Image]:
 
+    curPath = os.path.abspath(os.path.dirname(__file__))
+    checkpoint_dir = os.path.join(curPath, 'animeganv2/checkpoint/generator_Shinkai_weight')
+    print(checkpoint_dir)
+
+    test.test(checkpoint_dir, 'save_dir', image.name, True)
+
 
     return PIL.Image.open(image.name)