ajayarora1235 commited on
Commit
4b5e5e7
·
1 Parent(s): cd4126a

fix global vars

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -234,6 +234,9 @@ config = Config()
234
  logging.getLogger("numba").setLevel(logging.WARNING)
235
 
236
  hubert_model = None
 
 
 
237
 
238
  def load_hubert():
239
  global hubert_model
@@ -256,9 +259,7 @@ def load_hubert():
256
  # hubert_model = models[0]
257
 
258
  def load_voicecraft():
259
- global voicecraft_model
260
- global phn2num
261
- global voicecraft_config
262
 
263
  from lib.voicecraft.models import voicecraft
264
  voicecraft_name = "giga330M.pth"
@@ -1493,6 +1494,8 @@ def transcribe_btn_click(audio_choice):
1493
 
1494
  def run(seed, stop_repetition, sample_batch_size, left_margin, right_margin, codec_audio_sr, codec_sr, top_k, top_p,
1495
  temperature, kvcache, cutoff_value, target_transcript, silence_tokens, transcribed_text):
 
 
1496
  os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
1497
  os.environ["CUDA_VISIBLE_DEVICES"] = "0"
1498
  # take a look at demo/temp/mfa_alignment, decide which part of the audio to use as prompt
 
234
  logging.getLogger("numba").setLevel(logging.WARNING)
235
 
236
  hubert_model = None
237
+ voicecraft_model = None
238
+ voicecraft_config = None
239
+ phn2num = None
240
 
241
  def load_hubert():
242
  global hubert_model
 
259
  # hubert_model = models[0]
260
 
261
  def load_voicecraft():
262
+ global voicecraft_model, phn2num, voicecraft_config
 
 
263
 
264
  from lib.voicecraft.models import voicecraft
265
  voicecraft_name = "giga330M.pth"
 
1494
 
1495
  def run(seed, stop_repetition, sample_batch_size, left_margin, right_margin, codec_audio_sr, codec_sr, top_k, top_p,
1496
  temperature, kvcache, cutoff_value, target_transcript, silence_tokens, transcribed_text):
1497
+ global voicecraft_model, voicecraft_config, phn2num
1498
+
1499
  os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
1500
  os.environ["CUDA_VISIBLE_DEVICES"] = "0"
1501
  # take a look at demo/temp/mfa_alignment, decide which part of the audio to use as prompt