mineeuk commited on
Commit
824783f
·
1 Parent(s): 67d992f

fix: guard torchaudio.info patch with hasattr check for torchaudio 2.9+

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -10,7 +10,6 @@ import numpy as np
10
  import torchaudio
11
 
12
  _orig_torchaudio_load = torchaudio.load
13
- _orig_torchaudio_info = torchaudio.info
14
 
15
 
16
  def _patched_load(*args, **kwargs):
@@ -18,13 +17,15 @@ def _patched_load(*args, **kwargs):
18
  return _orig_torchaudio_load(*args, **kwargs)
19
 
20
 
21
- def _patched_info(*args, **kwargs):
22
- kwargs.setdefault("backend", "soundfile")
23
- return _orig_torchaudio_info(*args, **kwargs)
24
 
 
 
 
25
 
26
- torchaudio.load = _patched_load
27
- torchaudio.info = _patched_info
28
 
29
  from inference import inference
30
  from huggingface_hub import hf_hub_download
 
10
  import torchaudio
11
 
12
  _orig_torchaudio_load = torchaudio.load
 
13
 
14
 
15
  def _patched_load(*args, **kwargs):
 
17
  return _orig_torchaudio_load(*args, **kwargs)
18
 
19
 
20
+ torchaudio.load = _patched_load
21
+ if hasattr(torchaudio, "info"):
22
+ _orig_torchaudio_info = torchaudio.info
23
 
24
+ def _patched_info(*args, **kwargs):
25
+ kwargs.setdefault("backend", "soundfile")
26
+ return _orig_torchaudio_info(*args, **kwargs)
27
 
28
+ torchaudio.info = _patched_info
 
29
 
30
  from inference import inference
31
  from huggingface_hub import hf_hub_download