harp-dev commited on
Commit
3d82c4d
·
verified ·
1 Parent(s): 14fdb19

Declare the model's own sample rate and channel count

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,7 +12,7 @@ import librosa
12
  import torch
13
  import soundfile as sf
14
  from huggingface_hub import hf_hub_download
15
- from pyharp import ModelCard, build_endpoint
16
  from pyharp.tags import Subcategory
17
 
18
  from models.resunet import ResUNet30
@@ -80,7 +80,7 @@ model_card = ModelCard(
80
  )
81
  ),
82
  author="Xubo Liu, Qiuqiang Kong, Yan Zhao, Haohe Liu, Yi Yuan, Yuzhuo Liu, Rui Xia, Yuxuan Wang, Mark D. Plumbley, Wenwu Wang",
83
- tags=[Subcategory.TARGET_SOURCE_EXTRACTION, 'audio separation', 'text-queried', 'source separation'],
84
  )
85
 
86
 
 
12
  import torch
13
  import soundfile as sf
14
  from huggingface_hub import hf_hub_download
15
+ from pyharp import ModelCard, build_endpoint, SampleRate, Channels
16
  from pyharp.tags import Subcategory
17
 
18
  from models.resunet import ResUNet30
 
80
  )
81
  ),
82
  author="Xubo Liu, Qiuqiang Kong, Yan Zhao, Haohe Liu, Yi Yuan, Yuzhuo Liu, Rui Xia, Yuxuan Wang, Mark D. Plumbley, Wenwu Wang",
83
+ tags=[Subcategory.TARGET_SOURCE_EXTRACTION, SampleRate(32000), Channels(1), 'audio separation', 'text-queried', 'source separation'],
84
  )
85
 
86