"vscode:/vscode.git/clone" did not exist on "6217a6c2433ea05d1eeaa2c16f5a3e36446c3d41"
Commit 786da939 authored by myhloli's avatar myhloli
Browse files

feat(gui): update language options and default settings

- Remove unused 'layoutlmv3' model option
- Update language options to include new 'add_lang' list
- Set default language to 'ch' (Chinese)
- Comment out old 'all_lang' definition for future reference
parent ce212da1
...@@ -159,9 +159,12 @@ devanagari_lang = [ ...@@ -159,9 +159,12 @@ devanagari_lang = [
'sa', 'bgc' 'sa', 'bgc'
] ]
other_lang = ['ch', 'en', 'korean', 'japan', 'chinese_cht', 'ta', 'te', 'ka'] other_lang = ['ch', 'en', 'korean', 'japan', 'chinese_cht', 'ta', 'te', 'ka']
add_lang = ['latin', 'arabic', 'cyrillic', 'devanagari']
all_lang = ['', 'auto'] # all_lang = ['', 'auto']
all_lang.extend([*other_lang, *latin_lang, *arabic_lang, *cyrillic_lang, *devanagari_lang]) all_lang = []
# all_lang.extend([*other_lang, *latin_lang, *arabic_lang, *cyrillic_lang, *devanagari_lang])
all_lang.extend([*other_lang, *add_lang])
def to_pdf(file_path): def to_pdf(file_path):
...@@ -192,8 +195,8 @@ if __name__ == '__main__': ...@@ -192,8 +195,8 @@ if __name__ == '__main__':
file = gr.File(label='Please upload a PDF or image', file_types=['.pdf', '.png', '.jpeg', '.jpg']) file = gr.File(label='Please upload a PDF or image', file_types=['.pdf', '.png', '.jpeg', '.jpg'])
max_pages = gr.Slider(1, 20, 10, step=1, label='Max convert pages') max_pages = gr.Slider(1, 20, 10, step=1, label='Max convert pages')
with gr.Row(): with gr.Row():
layout_mode = gr.Dropdown(['layoutlmv3', 'doclayout_yolo'], label='Layout model', value='doclayout_yolo') layout_mode = gr.Dropdown(['doclayout_yolo'], label='Layout model', value='doclayout_yolo')
language = gr.Dropdown(all_lang, label='Language', value='auto') language = gr.Dropdown(all_lang, label='Language', value='ch')
with gr.Row(): with gr.Row():
formula_enable = gr.Checkbox(label='Enable formula recognition', value=True) formula_enable = gr.Checkbox(label='Enable formula recognition', value=True)
is_ocr = gr.Checkbox(label='Force enable OCR', value=False) is_ocr = gr.Checkbox(label='Force enable OCR', value=False)
......
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