Update app.py

app.py

@@ -25,6 +25,7 @@ def inference(
         "AnimeGANv2_Arcane": "arcane:v2",
         "AnimeGANv2_Test": "GeneratorV2_train_photo_Hayao.pt",
         "SummerWar": "GeneratorV2_train_photo_SummerWar.pt",
+        "Hetalia": "GeneratorV2_train_photo_Hetalia.pt",
     }[style]
     predictor = Predictor(
         weight,
@@ -54,6 +55,7 @@ gr.Interface(
                 'AnimeGANv2_Arcane',
                 'AnimeGANv2_Test',
                 'SummerWar',
+                'Hetalia',
             ],
             type="value",
             value='AnimeGANv2_Hayao',