Skip to content

GitLab

Commit 78238f39 authored by liukaiwen's avatar liukaiwen
Browse files

add table recognition and conversion to LaTeX

parent 3778cbc4
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
magic_pdf/model/pek_sub_modules/structeqtable/StructTableModel.py
View file @ 78238f39
...@@ -6,10 +6,10 @@ class StructTableModel: ...@@ -6,10 +6,10 @@ class StructTableModel:
self.model_path = model_path self.model_path = model_path
self.max_new_tokens = max_new_tokens # maximum output tokens length self.max_new_tokens = max_new_tokens # maximum output tokens length
self.max_time = max_time # timeout for processing in seconds self.max_time = max_time # timeout for processing in seconds
if device == 'cpu': if device == 'cuda':
self.model = StructTable(self.model_path, self.max_new_tokens, self.max_time)
else:
self.model = StructTable(self.model_path, self.max_new_tokens, self.max_time).cuda() self.model = StructTable(self.model_path, self.max_new_tokens, self.max_time).cuda()
else:
self.model = StructTable(self.model_path, self.max_new_tokens, self.max_time)
def image2latex(self, image) -> str: def image2latex(self, image) -> str:
# #
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment