Update app.py

app.py

@@ -2,19 +2,19 @@ from fastapi import FastAPI, File, UploadFile
 import librosa
 import numpy as np
 import shutil
-from funasr import AutoModel
-from funasr.utils.postprocess_utils import rich_transcription_postprocess
-
-from funasr.auto.auto_model import AutoModel
+import uvicorn
 import os
+from funasr import AutoModel
 
-
+# Khởi tạo FastAPI
 app = FastAPI()
 
-# Load mô hình SenseVoiceSmall
-model_dir = "FunAudioLLM/SenseVoiceSmall"
-
+# Tạo thư mục temp nếu chưa có
+if not os.path.exists("temp"):
+    os.makedirs("temp")
 
+# Load mô hình SenseVoiceSmall từ Hugging Face
+model_dir = "FunAudioLLM/SenseVoiceSmall"
 model = AutoModel(
     model=model_dir,
     vad_model="fsmn-vad",
@@ -52,3 +52,7 @@ async def detect_noise_api(file: UploadFile = File(...)):
 
     result = detect_noise(file_path)
     return {"noise_level": result}
+
+# Chạy FastAPI trên Hugging Face Spaces
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)