Initial

app.py

@@ -6,6 +6,7 @@ import asyncio
 import gradio as gr
 from omegaconf import OmegaConf
 from nemo.collections.asr.models.msdd_models import NeuralDiarizer
+from huggingface_hub import login
 
 # Local imports
 from src.audio.utils import Formatter
@@ -58,6 +59,8 @@ async def main(audio_file_path: str):
     device = config.runtime.device
     compute_type = config.runtime.compute_type
     os.environ["PYTORCH_CUDA_ALLOC_CONF"] = config.runtime.cuda_alloc_conf
+    hf_token = os.getenv("HF_TOKEN")
+    login(token=hf_token)
 
     # Initialize Classes
     dialogue_detector = DialogueDetecting(delete_original=True)