diff --git a/test_image_embedding_pipeline_yaml.py b/test_image_embedding_pipeline_yaml.py
index f0d3c2e..7bf7cd7 100644
--- a/test_image_embedding_pipeline_yaml.py
+++ b/test_image_embedding_pipeline_yaml.py
@@ -1,6 +1,4 @@
-import unittest
 from towhee import pipeline
-from config import DIMENSION, REPO_NAME, TEST_IMG
 
 
 def test_image_embedding_resnet50(repo_name='towhee/image-embedding-pipeline-template', img_path='./test_data/test.jpg', dimension=2048):