prithivMLmods commited on
Commit
eeee0c0
·
verified ·
1 Parent(s): f97523c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -103,7 +103,7 @@ def generate_image(model_name: str, text: str, image: Image.Image,
103
  elif model_name == "coreOCR-7B-050325-preview":
104
  processor = processor_x
105
  model = model_x
106
- elif model_name == "MonkeyOCR":
107
  processor = processor_g
108
  model = model_g
109
  else:
@@ -157,7 +157,7 @@ def generate_video(model_name: str, text: str, video_path: str,
157
  elif model_name == "coreOCR-7B-050325-preview":
158
  processor = processor_x
159
  model = model_x
160
- elif model_name == "MonkeyOCR":
161
  processor = processor_g
162
  model = model_g
163
  else:
@@ -258,7 +258,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
258
  with gr.Column():
259
  output = gr.Textbox(label="Output", interactive=False, lines=2, scale=2)
260
  model_choice = gr.Radio(
261
- choices=["docscopeOCR-7B-050425-exp", "MonkeyOCR", "coreOCR-7B-050325-preview"],
262
  label="Select Model",
263
  value="docscopeOCR-7B-050425-exp"
264
  )
 
103
  elif model_name == "coreOCR-7B-050325-preview":
104
  processor = processor_x
105
  model = model_x
106
+ elif model_name == "MonkeyOCR-Recognition":
107
  processor = processor_g
108
  model = model_g
109
  else:
 
157
  elif model_name == "coreOCR-7B-050325-preview":
158
  processor = processor_x
159
  model = model_x
160
+ elif model_name == "MonkeyOCR-Recognition":
161
  processor = processor_g
162
  model = model_g
163
  else:
 
258
  with gr.Column():
259
  output = gr.Textbox(label="Output", interactive=False, lines=2, scale=2)
260
  model_choice = gr.Radio(
261
+ choices=["docscopeOCR-7B-050425-exp", "MonkeyOCR-Recognition", "coreOCR-7B-050325-preview"],
262
  label="Select Model",
263
  value="docscopeOCR-7B-050425-exp"
264
  )