prithivMLmods commited on
Commit
cb335fc
·
verified ·
1 Parent(s): a40d85a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -66,8 +66,8 @@ model_g = Qwen2_5_VLForConditionalGeneration.from_pretrained(
66
  ).to(device).eval()
67
  #-----------------------------subfolder-----------------------------#
68
 
69
- # Load Perseus-Doc-vl-0712
70
- MODEL_ID_O = "prithivMLmods/Perseus-Doc-vl-0712"
71
  processor_o = AutoProcessor.from_pretrained(MODEL_ID_O, trust_remote_code=True)
72
  model_o = Qwen2_5_VLForConditionalGeneration.from_pretrained(
73
  MODEL_ID_O,
@@ -116,7 +116,7 @@ def generate_image(model_name: str, text: str, image: Image.Image,
116
  elif model_name == "MonkeyOCR-Recognition":
117
  processor = processor_g
118
  model = model_g
119
- elif model_name == "Perseus-Doc-vl-0712":
120
  processor = processor_o
121
  model = model_o
122
  else:
@@ -174,7 +174,7 @@ def generate_video(model_name: str, text: str, video_path: str,
174
  elif model_name == "MonkeyOCR-Recognition":
175
  processor = processor_g
176
  model = model_g
177
- elif model_name == "Perseus-Doc-vl-0712":
178
  processor = processor_o
179
  model = model_o
180
  else:
@@ -290,7 +290,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
290
  markdown_output = gr.Markdown(label="(Result.Md)")
291
 
292
  model_choice = gr.Radio(
293
- choices=["docscopeOCR-7B-050425-exp", "MonkeyOCR-Recognition", "coreOCR-7B-050325-preview", "Perseus-Doc-vl-0712"],
294
  label="Select Model",
295
  value="docscopeOCR-7B-050425-exp"
296
  )
 
66
  ).to(device).eval()
67
  #-----------------------------subfolder-----------------------------#
68
 
69
+ # Load Camel-Doc-OCR-080125
70
+ MODEL_ID_O = "prithivMLmods/Camel-Doc-OCR-080125"
71
  processor_o = AutoProcessor.from_pretrained(MODEL_ID_O, trust_remote_code=True)
72
  model_o = Qwen2_5_VLForConditionalGeneration.from_pretrained(
73
  MODEL_ID_O,
 
116
  elif model_name == "MonkeyOCR-Recognition":
117
  processor = processor_g
118
  model = model_g
119
+ elif model_name == "Camel-Doc-OCR-080125(v2)":
120
  processor = processor_o
121
  model = model_o
122
  else:
 
174
  elif model_name == "MonkeyOCR-Recognition":
175
  processor = processor_g
176
  model = model_g
177
+ elif model_name == "Camel-Doc-OCR-080125(v2)":
178
  processor = processor_o
179
  model = model_o
180
  else:
 
290
  markdown_output = gr.Markdown(label="(Result.Md)")
291
 
292
  model_choice = gr.Radio(
293
+ choices=["Camel-Doc-OCR-080125(v2)", "docscopeOCR-7B-050425-exp", "MonkeyOCR-Recognition", "coreOCR-7B-050325-preview"],
294
  label="Select Model",
295
  value="docscopeOCR-7B-050425-exp"
296
  )