Fix demo

Signed-off-by: smajumdar <[email protected]>

app.py

@@ -90,7 +90,7 @@ model_dict = {}
 for model_name in SUPPORTED_MODEL_NAMES:
     try:
         iface = gr.Interface.load(f'models/{model_name}')
-        model_dict[model_name] = model_dict
+        model_dict[model_name] = iface
     except:
         pass
 
@@ -369,8 +369,21 @@ def infer_audio(model_name: str, audio_file: str) -> str:
 
         if model is not None:
             # Use HF API for transcription
-            transcriptions = model(audio_file)
-            return transcriptions
+            try:
+                transcriptions = model(audio_file)
+                return transcriptions
+            except Exception as e:
+                transcriptions = ""
+                error = ""
+
+                error += (
+                    f"The model `{model_name}` is currently loading and cannot be used "
+                    f"for transcription.<br>"
+                    f"Please try another model or wait a few minutes."
+                )
+
+                return error
+
         else:
             error = (
                 f"Error:- Could not find model {model_name} in list of available models : "
@@ -422,8 +435,6 @@ def transcribe(microphone, audio_file, model_name):
             time_diff = end - start
 
     except Exception as e:
-        print(traceback.print_exc())
-
         transcriptions = ""
         warn_output = warn_output