Update app.py

app.py

@@ -3,12 +3,21 @@ import librosa
 import numpy as np
 import shutil
 from funasr import AutoModel
+from funasr.utils.postprocess_utils import rich_transcription_postprocess
 
 app = FastAPI()
 
 # Load mô hình SenseVoiceSmall
 model_dir = "FunAudioLLM/SenseVoiceSmall"
-model = AutoModel(model=model_dir, vad_model="fsmn-vad", vad_kwargs={"max_single_segment_time": 30000}, device="cpu", hub="hf")
+
+
+model = AutoModel(
+    model=model_dir,
+    vad_model="fsmn-vad",
+    vad_kwargs={"max_single_segment_time": 30000},
+    device="cuda:0",
+    hub="hf",
+)
 
 # Hàm tính RMS energy
 def calculate_rms_energy(audio_path):