完成了web，ray server，重构了代码

2024-02-08 13:48:34 +00:00
parent 07c4c3dc01
commit 04b99b8451
20 changed files with 245 additions and 57 deletions
--- a/src/start_web.sh
+++ b/src/start_web.sh
@@ -0,0 +1,9 @@
+#!/usr/bin/env bash
+set -exu
+
+export CHECKPOINT_DIR=/home/lhy/code/TeXify/src/models/ocr_model/model_checkpoint
+export TOKENIZER_DIR=/home/lhy/code/TeXify/src/models/tokenizer/roberta-tokenizer-550K
+export USE_CUDA=False
+export NUM_BEAM=3
+
+streamlit run web.py