Spaces:
Runtime error
Runtime error
from flask import Flask, request, jsonify | |
from flask_cors import CORS, cross_origin | |
from transformers import pipeline | |
import os | |
# Kiểm tra biến môi trường TRANSFORMERS_CACHE | |
print(f"TRANSFORMERS_CACHE: {os.getenv('TRANSFORMERS_CACHE')}") | |
app = Flask(__name__) | |
CORS(app) | |
app.config['CORS_HEADERS'] = 'Content-type' | |
# Tải mô hình Hugging Face | |
transcriber = pipeline(model="Phong1807/whisper-datn11") | |
def transcribe(): | |
audio_data = request.files['audio'] | |
transcription = transcriber(audio_data)['text'] | |
return jsonify({'transcription': transcription}) | |
if __name__ == '__main__': | |
app.run(host='0.0.0.0', port=7860) | |