Update app.py
Browse files
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',
|