prithivMLmods commited on
Commit
f6800b3
·
verified ·
1 Parent(s): 37b11c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -101,7 +101,7 @@ def generate_image(model_name: str, text: str, image: Image.Image,
101
  elif model_name == "docscopeOCR-7B-050425-exp":
102
  processor = processor_x
103
  model = model_x
104
- elif model_name == "Captioner-7B":
105
  processor = processor_z
106
  model = model_z
107
  elif model_name == "visionOCR-3B":
@@ -158,7 +158,7 @@ def generate_video(model_name: str, text: str, video_path: str,
158
  elif model_name == "docscopeOCR-7B-050425-exp":
159
  processor = processor_x
160
  model = model_x
161
- elif model_name == "Captioner-7B":
162
  processor = processor_z
163
  model = model_z
164
  elif model_name == "visionOCR-3B":
@@ -274,7 +274,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
274
  markdown_output = gr.Markdown()
275
 
276
  model_choice = gr.Radio(
277
- choices=["Cosmos-Reason1-7B", "docscopeOCR-7B-050425-exp", "Captioner-7B", "visionOCR-3B"],
278
  label="Select Model",
279
  value="Cosmos-Reason1-7B"
280
  )
 
101
  elif model_name == "docscopeOCR-7B-050425-exp":
102
  processor = processor_x
103
  model = model_x
104
+ elif model_name == "Captioner-7B-Qwen2.5VL":
105
  processor = processor_z
106
  model = model_z
107
  elif model_name == "visionOCR-3B":
 
158
  elif model_name == "docscopeOCR-7B-050425-exp":
159
  processor = processor_x
160
  model = model_x
161
+ elif model_name == "Captioner-7B-Qwen2.5VL":
162
  processor = processor_z
163
  model = model_z
164
  elif model_name == "visionOCR-3B":
 
274
  markdown_output = gr.Markdown()
275
 
276
  model_choice = gr.Radio(
277
+ choices=["Cosmos-Reason1-7B", "docscopeOCR-7B-050425-exp", "Captioner-7B-Qwen2.5VL", "visionOCR-3B"],
278
  label="Select Model",
279
  value="Cosmos-Reason1-7B"
280
  )