Create app.py

app.py

@@ -0,0 +1,53 @@
+import gradio as gr
+import torch
+from torchvision.transforms.functional import to_tensor, to_pil_image
+from PIL import Image
+import cv2
+import numpy as np
+
+# Load AnimeGANv2 pretrained generator (e.g., "hayao")
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+anime_model = torch.hub.load(
+    "bryandlee/animegan2-pytorch:main",
+    "generator",
+    pretrained="hayao",  # options: "hayao", "paprika", "face_paint_v1", "face_paint_v2"
+).to(device).eval()
+
+def apply_animegan(img: np.ndarray) -> np.ndarray:
+    pil = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
+    tensor = to_tensor(pil).unsqueeze(0).mul(2).sub(1).to(device)  # [-1,1]
+    with torch.no_grad():
+        out = anime_model(tensor)[0]
+    out = out.clamp(-1, 1).add(1).div(2)
+    pil_out = to_pil_image(out.cpu())
+    return cv2.cvtColor(np.array(pil_out), cv2.COLOR_RGB2BGR)
+
+def apply_cartoon_filter(img: np.ndarray) -> np.ndarray:
+    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+    gray = cv2.medianBlur(gray, 5)
+    edges = cv2.adaptiveThreshold(gray, 255,
+                                  cv2.ADAPTIVE_THRESH_MEAN_C,
+                                  cv2.THRESH_BINARY, 9, 9)
+    color = cv2.bilateralFilter(img, 9, 300, 300)
+    return cv2.bitwise_and(color, color, mask=edges)
+
+def cartoonizer(input_img, style):
+    img = cv2.cvtColor(np.array(input_img), cv2.COLOR_RGB2BGR)
+    if style == "Cartoon Filter":
+        out = apply_cartoon_filter(img)
+    elif style == "AnimeGANv2 (Hayao)":
+        out = apply_animegan(img)
+    else:
+        out = img
+    return Image.fromarray(cv2.cvtColor(out, cv2.COLOR_BGR2RGB))
+
+gr.Interface(
+    fn=cartoonizer,
+    inputs=[
+        gr.Image(type="pil", label="Upload Image"),
+        gr.Radio(["Cartoon Filter", "AnimeGANv2 (Hayao)"], label="Style")
+    ],
+    outputs=gr.Image(type="pil", label="Cartoonized Output"),
+    title="🎨 Cartoonizer with AnimeGANv2",
+    description="Upload any image and convert to general cartoon or Hayao‑style anime."
+).launch()