prithivMLmods commited on
Commit
f3eccec
·
verified ·
1 Parent(s): af174f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -142,7 +142,7 @@ def generate_image(model_name: str, text: str, image: Image.Image,
142
  processor, model = processor_x, model_x
143
  elif model_name == "olmOCR-7B-0225-preview":
144
  processor, model = processor_o, model_o
145
- elif model_name == "Typhoon-OCR-3B":
146
  processor, model = processor_t, model_t
147
  elif model_name == "Lumian-VLR-7B-Thinking":
148
  processor, model = processor_j, model_j
@@ -210,7 +210,7 @@ def generate_video(model_name: str, text: str, video_path: str,
210
  processor, model = processor_x, model_x
211
  elif model_name == "olmOCR-7B-0225-preview":
212
  processor, model = processor_o, model_o
213
- elif model_name == "Typhoon-OCR-3B":
214
  processor, model = processor_t, model_t
215
  elif model_name == "Lumian-VLR-7B-Thinking":
216
  processor, model = processor_j, model_j
@@ -298,8 +298,8 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
298
  output = gr.Textbox(label="Raw Output Stream", interactive=False, lines=4, show_copy_button=True)
299
  with gr.Accordion("(Result.md)", open=False):
300
  markdown_output = gr.Markdown(label="(Result.Md)")
301
- model_choice = gr.Dropdown(
302
- choices=[ "openbmb/MiniCPM-V-4", "Lumian-VLR-7B-Thinking", "DREX-062225-7B-exp", "olmOCR-7B-0225-preview", "Typhoon-OCR-3B"],
303
  label="Select Model",
304
  value="openbmb/MiniCPM-V-4"
305
  )
 
142
  processor, model = processor_x, model_x
143
  elif model_name == "olmOCR-7B-0225-preview":
144
  processor, model = processor_o, model_o
145
+ elif model_name == "Typhoon-OCR":
146
  processor, model = processor_t, model_t
147
  elif model_name == "Lumian-VLR-7B-Thinking":
148
  processor, model = processor_j, model_j
 
210
  processor, model = processor_x, model_x
211
  elif model_name == "olmOCR-7B-0225-preview":
212
  processor, model = processor_o, model_o
213
+ elif model_name == "Typhoon-OCR":
214
  processor, model = processor_t, model_t
215
  elif model_name == "Lumian-VLR-7B-Thinking":
216
  processor, model = processor_j, model_j
 
298
  output = gr.Textbox(label="Raw Output Stream", interactive=False, lines=4, show_copy_button=True)
299
  with gr.Accordion("(Result.md)", open=False):
300
  markdown_output = gr.Markdown(label="(Result.Md)")
301
+ model_choice = gr.Radio(
302
+ choices=[ "openbmb/MiniCPM-V-4", "Lumian-VLR-7B-Thinking", "Typhoon-OCR", "DREX-062225-7B-exp", "olmOCR-7B-0225-preview"],
303
  label="Select Model",
304
  value="openbmb/MiniCPM-V-4"
305
  )