Create app.py

app.py

@@ -0,0 +1,54 @@
+import os
+import cv2
+import gradio as gr
+from huggingface_hub import hf_hub_download
+
+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'),
+}
+
+
+def inference(img_path, model, focus_face=None):
+    print(img_path, model, focus_face)
+    mat, scale = load_image(img_path, focus_face)
+    output = convert(mat, model, scale)
+    save_path = f"output/out.{img_path.rsplit('.')[-1]}"
+    cv2.imwrite(save_path, output)
+    return output, save_path
+
+
+title = "AnimeGANv2: To produce your own animation 😶‍🌫️"
+description = r"""### 🔥Demo AnimeGANv2: To produce your own animation. To use it, simply upload your image.<br> 
+"""
+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>
+"""
+gr.Interface(
+    inference, [
+        gr.Image(type="filepath", label="Input image"),
+        gr.Dropdown([
+            'AnimeGANv2_Hayao',
+            'AnimeGANv2_Shinkai',
+            'AnimeGANv2_Paprika',
+            'AnimeGANv3_PortraitSketch',
+            'AnimeGANv3_JP_face',
+        ], 
+            type="value",
+            value='AnimeGANv3_PortraitSketch',
+            label='AnimeGAN Style'),
+        gr.Radio(['Yes', 'No'], type="value", value='No', label='Extract face'),
+    ], [
+        gr.Image(type="numpy", label="Output (The whole image)"),
+        gr.File(label="Download the output image")
+    ],
+    title=title,
+    description=description,
+    article=article,
+    allow_flagging="never").launch(debug=True)