ChaosTong commited on
Commit
d5baf32
·
verified ·
1 Parent(s): 4ac6a6c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -24,6 +24,7 @@ def inference(
24
  "AnimeGANv2_Shinkai": "shinkai:v2",
25
  "AnimeGANv2_Arcane": "arcane:v2",
26
  "AnimeGANv2_Test": "GeneratorV2_train_photo_Hayao.pt",
 
27
  }[style]
28
  predictor = Predictor(
29
  weight,
@@ -52,6 +53,7 @@ gr.Interface(
52
  'AnimeGANv2_Shinkai',
53
  'AnimeGANv2_Arcane',
54
  'AnimeGANv2_Test',
 
55
  ],
56
  type="value",
57
  value='AnimeGANv2_Hayao',
 
24
  "AnimeGANv2_Shinkai": "shinkai:v2",
25
  "AnimeGANv2_Arcane": "arcane:v2",
26
  "AnimeGANv2_Test": "GeneratorV2_train_photo_Hayao.pt",
27
+ "SummerWar": "GeneratorV2_train_photo_SummerWar.pt",
28
  }[style]
29
  predictor = Predictor(
30
  weight,
 
53
  'AnimeGANv2_Shinkai',
54
  'AnimeGANv2_Arcane',
55
  'AnimeGANv2_Test',
56
+ 'SummerWar',
57
  ],
58
  type="value",
59
  value='AnimeGANv2_Hayao',