Add universal model and updated texts.

app.py

@@ -40,28 +40,32 @@ def _getWindowingArray(window_size, fade_size):
 
 
 description = f'''
-texts
+This is unofficial space for audio restoration model Apollo: https://github.com/JusperLee/Apollo
 '''
 
 
 apollo_config = get_config('configs/apollo.yaml')
 apollo_vocal2_config = get_config('configs/config_apollo_vocal.yaml')
+apollo_uni_config = get_config('configs/config_apollo_uni.yaml')
 apollo_model = look2hear.models.BaseModel.from_pretrain('weights/apollo.bin', **apollo_config['model']).to(device)
 apollo_vocal = look2hear.models.BaseModel.from_pretrain('weights/apollo_vocal.bin', **apollo_config['model']).to(device)
 apollo_vocal2 = look2hear.models.BaseModel.from_pretrain('weights/apollo_vocal2.bin', **apollo_vocal2_config['model']).to(device)
+apollo_uni = look2hear.models.BaseModel.from_pretrain('weights/apollo_model_uni.ckpt', **apollo_uni_config['model']).to(device)
 
 
 
 models = {
    'apollo': apollo_model,
    'apollo_vocal': apollo_vocal,
-   'apollo_vocal2': apollo_vocal2
+   'apollo_vocal2': apollo_vocal2,
+   'apollo_uni': apollo_uni
 }
 
 choices = [
     ('MP3 restore', 'apollo'),
     ('Apollo vocal', 'apollo_vocal'),
-    ('Apollo vocal2', 'apollo_vocal2')
+    ('Apollo vocal2', 'apollo_vocal2'),
+    ('Apollo universal', 'apollo_uni')
 ]
 
 @spaces.GPU
@@ -151,7 +155,7 @@ if __name__ == "__main__":
         ],
         allow_flagging ='never',
         cache_examples=False,
-        title='Enchanser',
+        title='Apollo audio restoration',
         
     )
     i.queue(max_size=20, default_concurrency_limit=4)

configs/config_apollo_uni.yaml

@@ -0,0 +1,5 @@
+model:
+  sr: 44100
+  win: 20
+  feature_dim: 384
+  layer: 6

weights/apollo_model_uni.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:11433256eca90143d32f1108d156ac756fd0201fbc69e5f736def4062fb52186
+size 146867952