add pcen_cnn to models

BirdVox · Dec 1, 2018 · 1cd26a0 · 1cd26a0
1 parent e935a32
commit 1cd26a0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -10,7 +10,10 @@
     from urllib import urlretrieve
 
 model_dir = os.path.join('birdvoxdetect', 'models')
-suffixes = ['pcen_cnn_adaptive-threshold-T1800']
+suffixes = [
+    'pcen_cnn',
+    'pcen_cnn_adaptive-threshold-T1800'
+]
 weight_files = ['birdvoxdetect_{}.h5'.format(suffix) for suffix in suffixes]
 base_url = 'https://github.com/BirdVox/birdvoxdetect/raw/models/'