Spaces:

ChaosTong
/

AnimeGAN

Sleeping

ChaosTong commited on Dec 15, 2024

Commit

531be4a

verified ·

1 Parent(s): 13ed781

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,9 +20,9 @@ def inference(
     retain_color = retain_color
     weight = {
-        "AnimeGANv2_Hayao": "hayao:v2",
-        "AnimeGANv2_Shinkai": "shinkai:v2",
-        "AnimeGANv2_Arcane": "arcane:v2",
         "AnimeGANv2_Test": "GeneratorV2_train_photo_Hayao.pt",
         "SummerWar": "GeneratorV2_train_photo_SummerWar.pt",
         "Hetalia": "GeneratorV2_train_photo_Hetalia.pt",

     retain_color = retain_color
     weight = {
+        "AnimeGANv2_Hayao": "GeneratorV2_gldv2_Hayao.pt",
+        "AnimeGANv2_Shinkai": "GeneratorV2_gldv2_Shinkai.pt",
+        "AnimeGANv2_Arcane": "GeneratorV2_ffhq_Arcane_210624_e350.pt",
         "AnimeGANv2_Test": "GeneratorV2_train_photo_Hayao.pt",
         "SummerWar": "GeneratorV2_train_photo_SummerWar.pt",
         "Hetalia": "GeneratorV2_train_photo_Hetalia.pt",