diff --git a/gradio_demo.py b/gradio_demo.py
index 55e0cf9..527d0c6 100644
--- a/gradio_demo.py
+++ b/gradio_demo.py
@@ -153,7 +153,7 @@ def create_multi_modal_demo():
     return instruct_demo
 
 description = """
-# Hallucination Correction for MLLMs🔧
+# Woodpecker: Hallucination Correction for MLLMs🔧
 **Note**: Due to network restrictions, it is recommended that the size of the uploaded image be less than 1M.
 """
 
diff --git a/inference.py b/inference.py
index d0a138e..e984978 100644
--- a/inference.py
+++ b/inference.py
@@ -5,7 +5,7 @@
 
 
 if __name__ == '__main__':
-    parser = argparse.ArgumentParser(description='Hallucination Correction for MLLMs.')
+    parser = argparse.ArgumentParser(description="Code for 'Woodpecker: Hallucination Correction for MLLMs Hallucination Correction for MLLMs'.")
     parser.add_argument('--image-path', type=str, help="file path for the text to be corrected.")
     parser.add_argument('--text', type=str, help="text from MLLM to be corrected")
     parser.add_argument('--cache-dir', type=str, help="dir for caching intermediate image",
diff --git a/vis_corrector.py b/vis_corrector.py
index d138764..18569a7 100644
--- a/vis_corrector.py
+++ b/vis_corrector.py
@@ -7,7 +7,6 @@
 from models.refiner import Refiner
 from tqdm import tqdm
 from typing import List, Dict
-import time
 
 class Corrector:
     def __init__(self, args) -> None: