Fancellu commited on
Commit
2fddbf8
·
verified ·
1 Parent(s): dbd0f6a

Update app.py

Browse files

Need to monkey patch torch.load

Files changed (1) hide show
  1. app.py +11 -3
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(text="Generate Audio", type="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()