Update MiniMonkey.py

This commit is contained in:
echo840
2024-09-22 15:56:20 +08:00
committed by GitHub
parent 47af6ce6d4
commit eb7ecc408f

View File

@@ -164,7 +164,7 @@ def _get_args():
parser.add_argument("--image_folder", type=str, default="./OCRBench_Images")
parser.add_argument("--output_folder", type=str, default="./results")
parser.add_argument("--OCRBench_file", type=str, default="./OCRBench/OCRBench.json")
parser.add_argument("--model_path", type=str, default='mx262/MiniMokney')#TODO Set the address of your model's weights
parser.add_argument("--model_path", type=str, default='mx262/MiniMonkey')#TODO Set the address of your model's weights
parser.add_argument("--save_name", type=str, default="MiniMokney") #TODO Set the name of the JSON file you save in the output_folder.
parser.add_argument("--num_workers", type=int, default=1)
args = parser.parse_args()