Spaces:
Running
on
Zero
Running
on
Zero
update app
Browse files
app.py
CHANGED
|
@@ -138,7 +138,7 @@ css = """
|
|
| 138 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
| 139 |
print(f"Using Main Device: {device}")
|
| 140 |
|
| 141 |
-
QWEN_VL_MODEL_ID = "Qwen/Qwen2.5-VL-
|
| 142 |
print(f"Loading OCR Model: {QWEN_VL_MODEL_ID}...")
|
| 143 |
|
| 144 |
qwen_processor = AutoProcessor.from_pretrained(QWEN_VL_MODEL_ID, trust_remote_code=True)
|
|
@@ -335,7 +335,7 @@ example_image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
|
|
| 335 |
|
| 336 |
with gr.Blocks() as demo:
|
| 337 |
gr.Markdown("# **Vision-to-VibeVoice-en**", elem_id="main-title")
|
| 338 |
-
gr.Markdown("Perform vision-to-audio inference with [Qwen2.5VL](https://huggingface.co/Qwen/Qwen2.5-VL-
|
| 339 |
with gr.Row():
|
| 340 |
with gr.Column(scale=1):
|
| 341 |
gr.Markdown("### 1. Vision Input")
|
|
@@ -355,7 +355,7 @@ with gr.Blocks() as demo:
|
|
| 355 |
cfg_slider = gr.Slider(minimum=1.0, maximum=3.0, value=1.5, step=0.1, label="CFG Scale (Speech Fidelity)")
|
| 356 |
|
| 357 |
with gr.Accordion("Advanced Options", open=False):
|
| 358 |
-
max_new_tokens = gr.Slider(label="Max Tokens", minimum=128, maximum=4096, step=128, value=
|
| 359 |
temperature = gr.Slider(label="Temperature", minimum=0.1, maximum=2.0, step=0.1, value=0.1)
|
| 360 |
|
| 361 |
submit_btn = gr.Button("Generate Speech", variant="primary", size="lg")
|
|
|
|
| 138 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
| 139 |
print(f"Using Main Device: {device}")
|
| 140 |
|
| 141 |
+
QWEN_VL_MODEL_ID = "Qwen/Qwen2.5-VL-3B-Instruct"
|
| 142 |
print(f"Loading OCR Model: {QWEN_VL_MODEL_ID}...")
|
| 143 |
|
| 144 |
qwen_processor = AutoProcessor.from_pretrained(QWEN_VL_MODEL_ID, trust_remote_code=True)
|
|
|
|
| 335 |
|
| 336 |
with gr.Blocks() as demo:
|
| 337 |
gr.Markdown("# **Vision-to-VibeVoice-en**", elem_id="main-title")
|
| 338 |
+
gr.Markdown("Perform vision-to-audio inference with [Qwen2.5VL](https://huggingface.co/Qwen/Qwen2.5-VL-3B-Instruct) + [VibeVoice-Realtime-0.5B](https://huggingface.co/microsoft/VibeVoice-Realtime-0.5B).")
|
| 339 |
with gr.Row():
|
| 340 |
with gr.Column(scale=1):
|
| 341 |
gr.Markdown("### 1. Vision Input")
|
|
|
|
| 355 |
cfg_slider = gr.Slider(minimum=1.0, maximum=3.0, value=1.5, step=0.1, label="CFG Scale (Speech Fidelity)")
|
| 356 |
|
| 357 |
with gr.Accordion("Advanced Options", open=False):
|
| 358 |
+
max_new_tokens = gr.Slider(label="Max Tokens", minimum=128, maximum=4096, step=128, value=1024)
|
| 359 |
temperature = gr.Slider(label="Temperature", minimum=0.1, maximum=2.0, step=0.1, value=0.1)
|
| 360 |
|
| 361 |
submit_btn = gr.Button("Generate Speech", variant="primary", size="lg")
|