tudeplom commited on
Commit
2aa1fa1
·
verified ·
1 Parent(s): 8154745

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -4
app.py CHANGED
@@ -8,7 +8,7 @@ from flask import Flask, request, jsonify
8
  app = Flask(__name__)
9
 
10
  # Kiểm tra và load model Vosk
11
- MODEL_PATH = "model/vosk-model"
12
  if not os.path.exists(MODEL_PATH):
13
  raise Exception("❌ Model Vosk không tìm thấy!")
14
  print("✅ Đang tải model Vosk...")
@@ -20,10 +20,17 @@ rec = KaldiRecognizer(model, 16000)
20
  # API nhận dữ liệu âm thanh và trả kết quả nhận diện giọng nói
21
  @app.route('/stt', methods=['POST'])
22
  def handle_audio():
23
- audio_data = request.data # Nhận dữ liệu âm thanh từ client
 
 
 
 
 
 
 
24
  print(f"📩 Nhận dữ liệu âm thanh: {len(audio_data)} bytes")
25
-
26
- # Chuyển đổi dữ liệu âm thanh thành numpy array
27
  audio_np = np.frombuffer(audio_data, dtype=np.int16)
28
 
29
  # Nhận diện giọng nói
 
8
  app = Flask(__name__)
9
 
10
  # Kiểm tra và load model Vosk
11
+ MODEL_PATH = "model/vosk-model" # Thay thế với đường dẫn đến mô hình của bạn
12
  if not os.path.exists(MODEL_PATH):
13
  raise Exception("❌ Model Vosk không tìm thấy!")
14
  print("✅ Đang tải model Vosk...")
 
20
  # API nhận dữ liệu âm thanh và trả kết quả nhận diện giọng nói
21
  @app.route('/stt', methods=['POST'])
22
  def handle_audio():
23
+ # Kiểm tra xem tệp âm thanh không
24
+ if 'file' not in request.files:
25
+ return jsonify({"error": "Không có tệp âm thanh trong yêu cầu."}), 400
26
+
27
+ audio_file = request.files['file']
28
+
29
+ # Đọc dữ liệu âm thanh từ tệp
30
+ audio_data = audio_file.read()
31
  print(f"📩 Nhận dữ liệu âm thanh: {len(audio_data)} bytes")
32
+
33
+ # Chuyển đổi dữ liệu âm thanh thành numpy array (16-bit mono)
34
  audio_np = np.frombuffer(audio_data, dtype=np.int16)
35
 
36
  # Nhận diện giọng nói