Skip to content

Commit

Permalink
update code
Browse files Browse the repository at this point in the history
  • Loading branch information
xjtupanda committed Sep 27, 2023
1 parent 1a0a6fa commit e3fcac3
Show file tree
Hide file tree
Showing 3 changed files with 2 additions and 3 deletions.
2 changes: 1 addition & 1 deletion gradio_demo.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.
"""

Expand Down
2 changes: 1 addition & 1 deletion inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down
1 change: 0 additions & 1 deletion vis_corrector.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit e3fcac3

Please sign in to comment.