prithivMLmods commited on
Commit
b3c3e3e
·
verified ·
1 Parent(s): 78389d3

update bpe ocr (#2)

Browse files

- update bpe ocr (0cdfd6f4345141a20e3fee412cfbf0e894eadca5)

Files changed (1) hide show
  1. demo_inference/TokenizerOCR.ipynb +2 -2
demo_inference/TokenizerOCR.ipynb CHANGED
@@ -44,7 +44,7 @@
44
  "\n",
45
  "# Define model options\n",
46
  "MODEL_OPTIONS = {\n",
47
- " \"Tokenized-OCR\": \"prithivMLmods/Tokenized-OCR\",\n",
48
  "}\n",
49
  "\n",
50
  "# Preload models and processors into CUDA\n",
@@ -256,7 +256,7 @@
256
  " model_choice = gr.Dropdown(\n",
257
  " label=\"Model Selection\",\n",
258
  " choices=list(MODEL_OPTIONS.keys()),\n",
259
- " value=\"Tokenized-OCR\"\n",
260
  " )\n",
261
  " input_media = gr.File(\n",
262
  " label=\"Upload Image\", type=\"filepath\"\n",
 
44
  "\n",
45
  "# Define model options\n",
46
  "MODEL_OPTIONS = {\n",
47
+ " \"Bpe-vocab-n-OCR\": \"prithivMLmods/Bpe-vocab-n-OCR\",\n",
48
  "}\n",
49
  "\n",
50
  "# Preload models and processors into CUDA\n",
 
256
  " model_choice = gr.Dropdown(\n",
257
  " label=\"Model Selection\",\n",
258
  " choices=list(MODEL_OPTIONS.keys()),\n",
259
+ " value=\"Bpe-vocab-n-OCR\"\n",
260
  " )\n",
261
  " input_media = gr.File(\n",
262
  " label=\"Upload Image\", type=\"filepath\"\n",