初步修改完成，但仍然有问题

2024-03-27 04:54:49 +00:00
parent 6373e19132
commit dbf35fe9c4
3 changed files with 28 additions and 8 deletions
--- a/src/models/ocr_model/model/TexTeller.py
+++ b/src/models/ocr_model/model/TexTeller.py
@@ -41,6 +41,7 @@ class TexTeller(VisionEncoderDecoderModel):


 if __name__ == "__main__":
+    pause = 1
    # texteller = TexTeller()
    # from ..utils.inference import inference
    # model = TexTeller.from_pretrained('/home/lhy/code/TexTeller/src/models/ocr_model/model/ckpt')