1) 实现了文本-公式混排识别; 2) 重构了项目结构

2024-04-21 00:05:14 +08:00
parent 3746ddd427
commit f84168a00b
19 changed files with 753 additions and 296 deletions
--- a/src/server.py
+++ b/src/server.py
@@ -56,7 +56,7 @@ class TexTellerServer:
    def predict(self, image_nparray) -> str:
        return inference(
            self.model, self.tokenizer, [image_nparray],
-            inf_mode=self.inf_mode, num_beams=self.num_beams
+            accelerator=self.inf_mode, num_beams=self.num_beams
        )[0]