Update audio_pipe.py

audio_pipe.py

@@ -6,7 +6,6 @@ from typing import List, Tuple
 import numpy as np
 import torch
 # from app.pipelines import Pipeline
-from app.pipelines.utils import ARG_OVERRIDES_MAP
 from fairseq import hub_utils
 from fairseq.checkpoint_utils import load_model_ensemble_and_task_from_hf_hub
 from fairseq.models.speech_to_speech.hub_interface import S2SHubInterface
@@ -18,6 +17,12 @@ from fairseq.models.text_to_speech.hub_interface import (
 )
 from huggingface_hub import snapshot_download
 
+ARG_OVERRIDES_MAP = {
+    "facebook/xm_transformer_s2ut_800m-es-en-st-asr-bt_h1_2022": {
+        "config_yaml": "config.yaml",
+        "task": "speech_to_text",
+    }
+}
 
 class SpeechToSpeechPipeline():
     def __init__(self, model_id: str):