diff --git a/anime_transfer_gradio.py b/anime_transfer_gradio.py
index a7a29b6..a6dd78f 100644
--- a/anime_transfer_gradio.py
+++ b/anime_transfer_gradio.py
@@ -2,6 +2,7 @@ import gradio
 import numpy
 from PIL import Image
 from pathlib import Path
+from typing import NamedTuple
 
 from towhee.operator import Operator
 from towhee import pipeline