Spaces:
Sleeping
Sleeping
File size: 983 Bytes
1203715 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
from flask import Flask, request, jsonify
from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq
from transformers import pipeline
app = Flask(__name__)
model = AutoModelForSpeechSeq2Seq.from_pretrained("GitNazarov/whisper-small-pt-3-uz")
processor = AutoProcessor.from_pretrained("GitNazarov/whisper-small-pt-3-uz")
@app.route('/', methods=['GET'])
def index():
return jsonify({"message": "Welcome to whisper uz!"})
@app.route('/transcribe', methods=['POST'])
def transcribe():
data = request.json['data']
inputs = processor(data, return_tensors="pt", sampling_rate=16000, max_new_tokens=100)
input_features = inputs.input_features
generated_ids = model.generate(inputs=input_features)
transcription = processor.batch_decode(generated_ids, skip_special_tokens=True)
transcription = ''.join(transcription)
return str(transcription), {'Content-Type': 'application/json'}
if __name__ == '__main__':
app.run(host='0.0.0.0', port=7860) |