Spaces:
Runtime error
Runtime error
File size: 1,322 Bytes
94e37d8 b3aa369 94e37d8 8a65341 835eaac b3aa369 94e37d8 c0cca0a b3aa369 94e37d8 8a65341 b3aa369 8a65341 49829de 8a65341 835eaac 8a65341 b029c51 8a65341 b03e1fb b3aa369 835eaac 8a65341 b3aa369 835eaac 8a65341 |
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 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 |
import os
os.system("pip install git+https://github.com/openai/whisper.git")
import whisper
from flask import Flask, jsonify, request
import requests
import time
from transformers import pipeline
model = whisper.load_model("small")
pipe = pipeline(
"automatic-speech-recognition",
model="openai/whisper-small.en",
chunk_length_s=15,
device=model.device,
)
app = Flask(__name__)
app.config['TIMEOUT'] = 60 * 10 # 10 mins
@app.route("/")
def indexApi():
return jsonify({"output": "okay"})
@app.route("/run", methods=['POST'])
def runApi():
start_time = time.time()
audio_url = request.form.get("audio_url")
response = requests.get(audio_url)
if response.status_code == requests.codes.ok:
with open("audio.mp3", "wb") as f:
f.write(response.content)
else:
return jsonify({
"result": "Unable to save file, status code: {response.status_code}" ,
}), 400
audio = "audio.mp3"
audioOri = whisper.load_audio(audio)
prediction = pipe(audioOri)["text"]
end_time = time.time()
total_time = end_time - start_time
return jsonify({
"audio_url": audio_url,
"result": prediction,
"exec_time_sec": total_time
})
if __name__ == "__main__":
app.run(host="0.0.0.0", port=7860)
|