Update app.py

app.py

@@ -87,11 +87,11 @@ def convert(img, model, scale):
 os.makedirs('output', exist_ok=True)
 
 MODEL_PATH = {
-    "AnimeGANv2_Hayao": hf_hub_download('vumichien/AnimeGANv2_Hayao', 'AnimeGANv2_Hayao.onnx'),
-    "AnimeGANv2_Shinkai": hf_hub_download('vumichien/AnimeGANv2_Shinkai', 'AnimeGANv2_Shinkai.onnx'),
-    "AnimeGANv2_Paprika": hf_hub_download('vumichien/AnimeGANv2_Paprika', 'AnimeGANv2_Paprika.onnx'),
-    "AnimeGANv3_PortraitSketch": hf_hub_download('vumichien/AnimeGANv3_PortraitSketch', 'AnimeGANv3_PortraitSketch.onnx'),
-    "AnimeGANv3_JP_face": hf_hub_download('vumichien/AnimeGANv3_JP_face', 'AnimeGANv3_JP_face.onnx'),
+    "Hayao": hf_hub_download('vumichien/AnimeGANv2_Hayao', 'AnimeGANv2_Hayao.onnx'),
+    "Shinkai": hf_hub_download('vumichien/AnimeGANv2_Shinkai', 'AnimeGANv2_Shinkai.onnx'),
+    "Paprika": hf_hub_download('vumichien/AnimeGANv2_Paprika', 'AnimeGANv2_Paprika.onnx'),
+    "PortraitSketch": hf_hub_download('vumichien/AnimeGANv3_PortraitSketch', 'AnimeGANv3_PortraitSketch.onnx'),
+    "JP_face": hf_hub_download('vumichien/AnimeGANv3_JP_face', 'AnimeGANv3_JP_face.onnx'),
 }
 
 
@@ -106,34 +106,35 @@ def inference(img_path, model, focus_face=None):
 ### Layout ###
 
 title = "AnimeGANv2: To produce your own animation 😶‍🌫️"
-description = r"""### 🔥Demo AnimeGANv2: To produce your own animation. <br>
+description = r"""
+### 🔥Demo AnimeGANv2: To produce your own animation. <br>
 #### How to use:
-* Upload your image, 
-* Select the style, 
-* Choice of whether to extract the face.(Warning: Yes if there is a face in the image)
+* 1. Upload your image
+* 2. Select the style (For human: PortraitSketch, JP_face; For scene: Hayao, Shinkai, Paprika)
+* 3. Choice of whether to extract the face.(Warning: Yes if there is a face in the image)
 """
 article = r"""
 <center><img src='https://visitor-badge.glitch.me/badge?page_id=AnimeGAN_demo&left_color=green&right_color=blue' alt='visitor badge'></center>
 <center><a href='https://github.com/TachibanaYoshino/AnimeGANv3' target='_blank'>Github Repo</a></center>
 """
-examples=[['sample1.jpg', 'AnimeGANv3_PortraitSketch', "Yes"], 
-          ['sample2.jpg', 'AnimeGANv3_PortraitSketch', "No"],
-          ['sample3.jpg', 'AnimeGANv2_Hayao', "No"], 
-          ['sample4.jpeg', 'AnimeGANv2_Shinkai', "No"],
-          ['sample5.jpg', 'AnimeGANv2_Paprika', "No"], 
-          ['sample6.jpeg', 'AnimeGANv3_JP_face', "No"]]
+examples=[['sample1.jpg', 'PortraitSketch', "Yes"], 
+          ['sample2.jpg', 'PortraitSketch', "No"],
+          ['sample3.jpg', 'Hayao', "No"], 
+          ['sample4.jpeg', 'Shinkai', "No"],
+          ['sample5.jpg', 'Paprika', "No"], 
+          ['sample6.jpeg', 'JP_face', "No"]]
 gr.Interface(
     inference, [
         gr.Image(type="filepath", label="Input image"),
         gr.Dropdown([
-            'AnimeGANv2_Hayao',
-            'AnimeGANv2_Shinkai',
-            'AnimeGANv2_Paprika',
-            'AnimeGANv3_PortraitSketch',
-            'AnimeGANv3_JP_face',
+            'Hayao',
+            'Shinkai',
+            'Paprika',
+            'PortraitSketch',
+            'JP_face',
         ], 
             type="value",
-            value='AnimeGANv3_PortraitSketch',
+            value='PortraitSketch',
             label='AnimeGAN Style'),
         gr.Radio(['Yes', 'No'], type="value", value='No', label='Extract face'),
     ], [