修复了merge pre_release分支后导致参数名不一致的bug

2024-04-17 14:47:58 +00:00
parent 891a9c310a
commit 5a9138026f
2 changed files with 2 additions and 2 deletions
--- a/src/models/ocr_model/utils/inference.py
+++ b/src/models/ocr_model/utils/inference.py
@@ -13,7 +13,7 @@ from models.globals import MAX_TOKEN_SIZE
 def inference(
    model: TexTeller, 
    tokenizer: RobertaTokenizerFast,
-    imgs_path: Union[List[str], List[np.ndarray]], 
+    imgs: Union[List[str], List[np.ndarray]], 
    inf_mode: str = 'cpu',
    num_beams: int = 1,
 ) -> List[str]: