YogeLiu/TexTeller
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
三洋三洋
2024-04-17 14:47:58 +00:00
parent a064b7dbb0
commit 7883d3c07f
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/models/ocr_model/utils/inference.py
View File

@@ -13,7 +13,7 @@ from models.globals import MAX_TOKEN_SIZE
def inference( def inference(
model: TexTeller, model: TexTeller,
tokenizer: RobertaTokenizerFast, tokenizer: RobertaTokenizerFast,
imgs_path: Union[List[str], List[np.ndarray]], imgs: Union[List[str], List[np.ndarray]],
inf_mode: str = 'cpu', inf_mode: str = 'cpu',
num_beams: int = 1, num_beams: int = 1,
) -> List[str]: ) -> List[str]:

2
src/start_web.sh
View File

@@ -1,7 +1,7 @@
#!/usr/bin/env bash #!/usr/bin/env bash
set -exu 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" export TOKENIZER_DIR="default"
streamlit run web.py streamlit run web.py