livebook-dev/livebook
1
1
Fork 0
mirror of https://github.com/livebook-dev/livebook.git synced 2024-11-10 09:03:02 +08:00
Code Issues Projects Releases Packages Wiki Activity

Update transcription drop action for audio files

Browse source
This commit is contained in:
Jonatan Kłosko 2023-09-15 13:40:24 +07:00
parent b8fc4eefb5
commit 4d42202c83
1 changed files with 12 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
lib/livebook/runtime/definitions.ex
View file

@ -351,20 +351,28 @@ defmodule Livebook.Runtime.Definitions do
#{if windows? do
"""
serving = Bumblebee.Audio.speech_to_text(model_info, featurizer, tokenizer, generation_config)\
serving = Bumblebee.Audio.speech_to_text_whisper(model_info, featurizer, tokenizer, generation_config,
chunk_num_seconds: 30,
timestamps: :segments,
compile: [batch_size: 4]
)\
"""
else
"""
serving =
Bumblebee.Audio.speech_to_text(model_info, featurizer, tokenizer, generation_config,
compile: [batch_size: 1],
Bumblebee.Audio.speech_to_text_whisper(model_info, featurizer, tokenizer, generation_config,
chunk_num_seconds: 30,
timestamps: :segments,
compile: [batch_size: 4],
defn_options: [compiler: EXLA]
)\
"""
end}
path = Kino.FS.file_path("{{NAME}}")
output = Nx.Serving.run(serving, {:file, path})\
output = Nx.Serving.run(serving, {:file, path})
# output.chunks |> Enum.map_join(& &1.text) |> String.trim()\
""",
packages: [kino_bumblebee, nx_backend_package]
},