Update files

2024-02-12 11:40:51 +00:00
parent fa6bcda721
commit 9bc165f955
4 changed files with 178 additions and 15 deletions
--- a/src/start_web.sh
+++ b/src/start_web.sh
@@ -1,9 +1,9 @@
 #!/usr/bin/env bash
 set -exu

-export CHECKPOINT_DIR="OleehyO/TexTeller"
-export TOKENIZER_DIR="OleehyO/TexTeller"
-export USE_CUDA=True  # True or False (case-sensitive)
-export NUM_BEAM=5
+export CHECKPOINT_DIR="default"
+export TOKENIZER_DIR="default"
+export USE_CUDA=False  # True or False (case-sensitive)
+export NUM_BEAM=1

 streamlit run web.py