prithivMLmods commited on
Commit
2497cdf
·
verified ·
1 Parent(s): 9050c4f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -43,7 +43,7 @@ model_x = Qwen2_5_VLForConditionalGeneration.from_pretrained(
43
  torch_dtype=torch.float16).to(device).eval()
44
 
45
  # Load prithivMLmods/WR30a-Deep-7B-0711
46
- MODEL_ID_T = "NCSOFT/VARCO-VISION-2.0-14B"
47
  processor_t = AutoProcessor.from_pretrained(MODEL_ID_T, trust_remote_code=True)
48
  model_t = LlavaOnevisionForConditionalGeneration.from_pretrained(
49
  MODEL_ID_T, trust_remote_code=True,
@@ -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-14B":
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-14B":
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-14B",
319
  "ViGaL-7B", "Visionary-R1-3B"
320
  ],
321
  label="Select Model",
@@ -344,6 +344,6 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
344
  repetition_penalty
345
  ],
346
  outputs=[output, markdown_output])
347
-
348
  if __name__ == "__main__":
349
  demo.queue(max_size=30).launch(share=True, mcp_server=True, ssr_mode=False, show_error=True)
 
43
  torch_dtype=torch.float16).to(device).eval()
44
 
45
  # Load prithivMLmods/WR30a-Deep-7B-0711
46
+ MODEL_ID_T = "NCSOFT/VARCO-VISION-2.0-1.7B-OCR"
47
  processor_t = AutoProcessor.from_pretrained(MODEL_ID_T, trust_remote_code=True)
48
  model_t = LlavaOnevisionForConditionalGeneration.from_pretrained(
49
  MODEL_ID_T, trust_remote_code=True,
 
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
  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
  #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",
 
344
  repetition_penalty
345
  ],
346
  outputs=[output, markdown_output])
347
+
348
  if __name__ == "__main__":
349
  demo.queue(max_size=30).launch(share=True, mcp_server=True, ssr_mode=False, show_error=True)