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()
|