Fix model select

app.py

@@ -47,12 +47,18 @@ texts
 apollo_config = get_config('configs/apollo.yaml')
 apollo_model = look2hear.models.BaseModel.from_pretrain('weights/apollo.bin', **apollo_config['model']).to(device)
 
-models = [
-    ('MP3 restore', apollo_model)
+models = {
+   'apollo':  apollo_model
+}
+
+choices = [
+    ('MP3 restore', 'apollo')
 ]
 
 @spaces.GPU
-def enchance(model, audio):
+def enchance(choice, audio):
+    print(choice)
+    model = models[choice]
     test_data, samplerate = load_audio(audio)
     C = 10 * samplerate  # chunk_size seconds to samples
     N = 2
@@ -122,7 +128,7 @@ if __name__ == "__main__":
         fn=enchance,
         description=description,
         inputs=[
-            gr.Dropdown(label="Model", choices=models, value=models[0]),
+            gr.Dropdown(label="Model", choices=choices, value=choices[0]),
             gr.Audio(label="Input Audio:", interactive=True, type='filepath', max_length=300, waveform_options={'waveform_progress_color': '#3C82F6'}),
         ],
         outputs=[