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

update README

Browse Source
This commit is contained in:
三洋三洋
2024-02-12 08:41:33 +00:00
parent b4962bfa98
commit 6e2e45a8d6
5 changed files with 590 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/models/globals.py
View File

@@ -2,9 +2,6 @@
IMAGE_MEAN = 0.9545467
IMAGE_STD = 0.15394445
# Density value for pdf to image conversion
TEXTELL_DENSITY = 200
# Vocabulary size for TexTeller
VOCAB_SIZE = 10000

4
src/start_web.sh
View File

@@ -3,7 +3,7 @@ set -exu
export CHECKPOINT_DIR="OleehyO/TexTeller"
export TOKENIZER_DIR="OleehyO/TexTeller"
export USE_CUDA=False # True or False (case-sensitive)
export NUM_BEAM=1
export USE_CUDA=True # True or False (case-sensitive)
export NUM_BEAM=5
streamlit run web.py