fix model loading

app.py

@@ -2,9 +2,18 @@ import gradio as gr
 from audiosr import super_resolution, build_model
 import spaces
 
+
+audiosr = build_model(model_name='basic')
+audiosr_speech = build_model(model_name='speech')
+
+models = {
+    'basic': audiosr,
+    'speech': audiosr_speech
+}
+
 @spaces.GPU
 def inference(audio_file, model_name, guidance_scale, ddim_steps):
-    audiosr = build_model(model_name=model_name)
+    audiosr = models[model_name]
     
     waveform = super_resolution(
         audiosr,

requirements.txt

@@ -9,4 +9,5 @@ soundfile
 progressbar
 librosa
 audiosr
-unidecode
+unidecode
+matplotlib