prithivMLmods commited on
Commit
9b5dfc8
·
verified ·
1 Parent(s): 2497cdf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -100,7 +100,7 @@ def generate_image(model_name: str,
100
  elif model_name == "Visionary-R1-3B":
101
  processor = processor_o
102
  model = model_o
103
- elif model_name == "VARCO-VISION-2.0-1.7B-OCR":
104
  processor = processor_t
105
  model = model_t
106
  else:
@@ -164,7 +164,7 @@ def generate_video(model_name: str,
164
  elif model_name == "Visionary-R1-3B":
165
  processor = processor_o
166
  model = model_o
167
- elif model_name == "VARCO-VISION-2.0-1.7B-OCR":
168
  processor = processor_t
169
  model = model_t
170
  else:
@@ -315,7 +315,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
315
  #download_btn = gr.Button("Download Result.md")
316
 
317
  model_choice = gr.Radio(choices=[
318
- "Camel-Doc-OCR-080125(v2)", "VARCO-VISION-2.0-1.7B-OCR",
319
  "ViGaL-7B", "Visionary-R1-3B"
320
  ],
321
  label="Select Model",
 
100
  elif model_name == "Visionary-R1-3B":
101
  processor = processor_o
102
  model = model_o
103
+ elif model_name == "Varco-Vision-2.0-OCR":
104
  processor = processor_t
105
  model = model_t
106
  else:
 
164
  elif model_name == "Visionary-R1-3B":
165
  processor = processor_o
166
  model = model_o
167
+ elif model_name == "Varco-Vision-2.0-OCR":
168
  processor = processor_t
169
  model = model_t
170
  else:
 
315
  #download_btn = gr.Button("Download Result.md")
316
 
317
  model_choice = gr.Radio(choices=[
318
+ "Camel-Doc-OCR-080125(v2)", "Varco-Vision-2.0-OCR",
319
  "ViGaL-7B", "Visionary-R1-3B"
320
  ],
321
  label="Select Model",