update code

BradyFU · Sep 27, 2023 · e3fcac3 · e3fcac3
1 parent 1a0a6fa
commit e3fcac3
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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: