Update lib/kino_bumblebee/task_cell.ex

Co-authored-by: José Valim <jose.valim@gmail.com>
livebook-dev · Sep 14, 2023 · f71c9e6 · f71c9e6
1 parent 975c75f
commit f71c9e6
Showing 1 changed file with 4 additions and 15 deletions.
diff --git a/lib/kino_bumblebee/task_cell.ex b/lib/kino_bumblebee/task_cell.ex
@@ -1321,21 +1321,10 @@ defmodule KinoBumblebee.TaskCell do
             for chunk <- Nx.Serving.run(serving, audio) do
               [start_mark, end_mark] =
                 for seconds <- [chunk.start_timestamp_seconds, chunk.end_timestamp_seconds] do
-                  seconds = round(seconds)
-
-                  minutes =
-                    seconds
-                    |> div(60)
-                    |> Integer.to_string()
-                    |> String.pad_leading(2, "0")
-
-                  seconds =
-                    seconds
-                    |> rem(60)
-                    |> Integer.to_string()
-                    |> String.pad_leading(2, "0")
-
-                  minutes <> ":" <> seconds
+                  seconds
+                  |> round()
+                  |> Time.from_seconds_after_midnight()
+                  |> Time.to_string()
                 end
 
               text = "#{start_mark}-#{end_mark}: #{chunk.text}"