Update app.py

app.py

@@ -16,7 +16,7 @@ import plotly.graph_objects as go
 
 ASR_EXAMPLES = [("assets/bach.wav")]
 
-aes_predictor = aes_infer.initialize_model(None)
+aes_predictor = aes_infer.initialize_predictor()
 
 
 def transcribe(audio_data=None):
@@ -41,9 +41,8 @@ def transcribe(audio_data=None):
             )
         audio_samples, sr = torchaudio.load(audio_data)
 
-    transcription = json.loads(
-        aes_predictor.forward([{"path": audio_samples, "sample_rate": sr}])[0]
-    )
+    # run prediction
+    transcription = aes_predictor.forward([{"path": audio_samples, "sample_rate": sr}])[0]
 
     # reorder like paper figures
     transcription = {
@@ -92,10 +91,6 @@ main_interface = gr.Interface(
     allow_flagging="never",
 )
 
-disclaimer = """
-## Disclaimer
-"""
-
 with gr.Blocks() as demo:
     gr.HTML(
         """
@@ -126,8 +121,6 @@ with gr.Blocks() as demo:
            """
     )
 
-    # with gr.Row():
-    #     gr.Markdown(disclaimer)
 
 if __name__ == "__main__":
     demo.queue()