Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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-
|
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 == "
|
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 == "
|
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)", "
|
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)
|