prithivMLmods commited on
Commit
94df60b
·
verified ·
1 Parent(s): 80a2981

update app

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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-7B-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,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-7B-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,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=2048)
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")