Update app.py
Browse filesNeed to monkey patch torch.load
app.py
CHANGED
@@ -1,5 +1,14 @@
|
|
1 |
import numpy as np
|
2 |
import gradio as gr
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
from bark import SAMPLE_RATE, generate_audio, preload_models
|
4 |
from bark.generation import SUPPORTED_LANGS
|
5 |
from share_btn import community_icon_html, loading_icon_html, share_js
|
@@ -173,7 +182,7 @@ with gr.Blocks(css=css) as block:
|
|
173 |
label="Input Text", lines=2, value=default_text, elem_id="input_text")
|
174 |
options = gr.Dropdown(
|
175 |
AVAILABLE_PROMPTS, value="Speaker 1 (en)", label="Acoustic Prompt", elem_id="speaker_option")
|
176 |
-
run_button = gr.Button(
|
177 |
with gr.Column():
|
178 |
audio_out = gr.Audio(label="Generated Audio",
|
179 |
type="numpy", elem_id="audio_out")
|
@@ -183,7 +192,6 @@ with gr.Blocks(css=css) as block:
|
|
183 |
loading_icon = gr.HTML(loading_icon_html)
|
184 |
share_button = gr.Button(
|
185 |
"Share to community", elem_id="share-btn")
|
186 |
-
share_button.click(None, [], [], _js=share_js)
|
187 |
inputs = [input_text, options]
|
188 |
outputs = [audio_out]
|
189 |
gr.Examples(examples=examples, fn=gen_tts, inputs=inputs,
|
@@ -194,4 +202,4 @@ with gr.Blocks(css=css) as block:
|
|
194 |
fn=lambda: gr.update(visible=True), inputs=None, outputs=share_row, queue=False)
|
195 |
|
196 |
block.queue()
|
197 |
-
block.launch()
|
|
|
1 |
import numpy as np
|
2 |
import gradio as gr
|
3 |
+
import torch
|
4 |
+
|
5 |
+
# Monkey patch torch.load to disable weights_only
|
6 |
+
original_load = torch.load
|
7 |
+
def patched_load(*args, **kwargs):
|
8 |
+
kwargs.setdefault('weights_only', False)
|
9 |
+
return original_load(*args, **kwargs)
|
10 |
+
torch.load = patched_load
|
11 |
+
|
12 |
from bark import SAMPLE_RATE, generate_audio, preload_models
|
13 |
from bark.generation import SUPPORTED_LANGS
|
14 |
from share_btn import community_icon_html, loading_icon_html, share_js
|
|
|
182 |
label="Input Text", lines=2, value=default_text, elem_id="input_text")
|
183 |
options = gr.Dropdown(
|
184 |
AVAILABLE_PROMPTS, value="Speaker 1 (en)", label="Acoustic Prompt", elem_id="speaker_option")
|
185 |
+
run_button = gr.Button(value="Generate Audio")
|
186 |
with gr.Column():
|
187 |
audio_out = gr.Audio(label="Generated Audio",
|
188 |
type="numpy", elem_id="audio_out")
|
|
|
192 |
loading_icon = gr.HTML(loading_icon_html)
|
193 |
share_button = gr.Button(
|
194 |
"Share to community", elem_id="share-btn")
|
|
|
195 |
inputs = [input_text, options]
|
196 |
outputs = [audio_out]
|
197 |
gr.Examples(examples=examples, fn=gen_tts, inputs=inputs,
|
|
|
202 |
fn=lambda: gr.update(visible=True), inputs=None, outputs=share_row, queue=False)
|
203 |
|
204 |
block.queue()
|
205 |
+
block.launch()
|