breezedeus commited on
Commit
84028a2
·
1 Parent(s): 0aff688

update copyright year to 2025, update Pix2Text version to 1.1.4, and modify model dropdown options

Browse files
Files changed (2) hide show
  1. app.py +6 -24
  2. requirements.txt +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  # coding: utf-8
2
  # [Pix2Text](https://github.com/breezedeus/pix2text): an Open-Source Alternative to Mathpix.
3
- # Copyright (C) 2022-2024, [Breezedeus](https://www.breezedeus.com).
4
 
5
  import os
6
  import json
@@ -31,24 +31,6 @@ OUTPUT_RESULT_DIR = Path('./output-results')
31
  OUTPUT_RESULT_DIR.mkdir(exist_ok=True)
32
 
33
 
34
- def prepare_mfd_model():
35
- target_fp = './yolov7-model/mfd-yolov7-epoch224-20230613.pt'
36
- if os.path.exists(target_fp):
37
- return target_fp
38
- HF_TOKEN = os.environ.get('HF_TOKEN')
39
- local_path = hf_hub_download(
40
- repo_id='breezedeus/paid-models',
41
- subfolder='cnstd/1.2',
42
- filename='yolov7-model-20230613.zip',
43
- repo_type="model",
44
- cache_dir='./',
45
- token=HF_TOKEN,
46
- )
47
- with zipfile.ZipFile(local_path) as zf:
48
- zf.extractall('./')
49
- return target_fp
50
-
51
-
52
  def get_p2t_model(lan_list: list, mfd_model_name: str, mfr_model_name: str):
53
  mfd_config = dict(model_name=mfd_model_name, model_backend='onnx') # 声明 MFD 的初始化参数
54
  formula_config = dict(model_name=mfr_model_name, model_backend='onnx') # 声明 MFR 的初始化参数
@@ -172,7 +154,7 @@ If useful, please help to **star 🌟 [Pix2Text](https://github.com/breezedeus/p
172
 
173
  with gr.Blocks() as demo:
174
  gr.HTML(
175
- f'<h1 style="text-align: center; margin-bottom: 1rem;"><a href="https://github.com/breezedeus/pix2text" target="_blank">Pix2Text V1.1.1</a>{title}</h1>'
176
  )
177
  with gr.Row(equal_height=False):
178
  with gr.Column(min_width=200, variant='panel', scale=3):
@@ -186,13 +168,13 @@ If useful, please help to **star 🌟 [Pix2Text](https://github.com/breezedeus/p
186
  )
187
  mfd_model_name = gr.Dropdown(
188
  label='MFD Models',
189
- choices=['mfd', 'mfd-advanced', 'mfd-pro'],
190
- value='mfd-pro',
191
  )
192
  mfr_model_name = gr.Dropdown(
193
  label='MFR Models',
194
- choices=['mfr', 'mfr-pro', 'mfr-plus'],
195
- value='mfr-pro',
196
  )
197
  rec_type = gr.Dropdown(
198
  label='file_type',
 
1
  # coding: utf-8
2
  # [Pix2Text](https://github.com/breezedeus/pix2text): an Open-Source Alternative to Mathpix.
3
+ # Copyright (C) 2022-2025, [Breezedeus](https://www.breezedeus.com).
4
 
5
  import os
6
  import json
 
31
  OUTPUT_RESULT_DIR.mkdir(exist_ok=True)
32
 
33
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
34
  def get_p2t_model(lan_list: list, mfd_model_name: str, mfr_model_name: str):
35
  mfd_config = dict(model_name=mfd_model_name, model_backend='onnx') # 声明 MFD 的初始化参数
36
  formula_config = dict(model_name=mfr_model_name, model_backend='onnx') # 声明 MFR 的初始化参数
 
154
 
155
  with gr.Blocks() as demo:
156
  gr.HTML(
157
+ f'<h1 style="text-align: center; margin-bottom: 1rem;"><a href="https://github.com/breezedeus/pix2text" target="_blank">Pix2Text V1.1.4</a>{title}</h1>'
158
  )
159
  with gr.Row(equal_height=False):
160
  with gr.Column(min_width=200, variant='panel', scale=3):
 
168
  )
169
  mfd_model_name = gr.Dropdown(
170
  label='MFD Models',
171
+ choices=['mfd-1.5', 'mfd-advanced-1.5', 'mfd-pro-1.5', 'mfd', 'mfd-advanced', 'mfd-pro'],
172
+ value='mfd-pro-1.5',
173
  )
174
  mfr_model_name = gr.Dropdown(
175
  label='MFR Models',
176
+ choices=['mfr-1.5', 'mfr-pro-1.5', 'mfr', 'mfr-pro', 'mfr-plus'],
177
+ value='mfr-pro-1.5',
178
  )
179
  rec_type = gr.Dropdown(
180
  label='file_type',
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
  --extra-index-url https://pypi.org/simple
2
 
3
  pyyaml
4
- pix2text[multilingual]>=1.1.1
 
1
  --extra-index-url https://pypi.org/simple
2
 
3
  pyyaml
4
+ pix2text[multilingual]>=1.1.4