修复了merge pre_release分支后导致参数名不一致的bug
This commit is contained in:
@@ -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]:
|
||||
|
||||
@@ -1,7 +1,7 @@
|
||||
#!/usr/bin/env bash
|
||||
set -exu
|
||||
|
||||
export CHECKPOINT_DIR="default"
|
||||
export CHECKPOINT_DIR="/home/lhy/code/TexTeller/src/models/ocr_model/train/train_result/TexTellerv3/checkpoint-648000"
|
||||
export TOKENIZER_DIR="default"
|
||||
|
||||
streamlit run web.py
|
||||
|
||||
Reference in New Issue
Block a user