From 240f4abe6ff538a6627f1cbcfe989c1b22e30b87 Mon Sep 17 00:00:00 2001
From: "junjie.jiang" <junjie.jiang@zilliz.com>
Date: Thu, 16 Mar 2023 19:50:17 +0800
Subject: [PATCH] update

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
---
 video_decoder.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/video_decoder.py b/video_decoder.py
index 114a82d..4c0b57c 100644
--- a/video_decoder.py
+++ b/video_decoder.py
@@ -3,12 +3,12 @@ import av
 import numpy as np
 from towhee.operator.base import PyOperator
 
-from cpu_decode import PyAVDecode
+from .cpu_decode import PyAVDecode
 
 logger = logging.getLogger()
 
 try:
-    from gpu_decode import VPFDecode
+    from .gpu_decode import VPFDecode
 except Exception:
     logger.error('Import GPUDecoder failed, use CPU decode')
     VPFDecode = PyAVDecode