mix format

cocoa-xu · Jan 24, 2024 · bca53f8 · bca53f8
1 parent d4f9450
commit bca53f8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/tflite_elixir/high-level-api/mobile_bert.ex b/lib/tflite_elixir/high-level-api/mobile_bert.ex
@@ -85,6 +85,7 @@ defmodule TFLiteElixir.MobileBert do
     content_words = String.split(content)
 
     content_tokens = Enum.map(content_words, &FullTokenizer.tokenize(&1, true, vocab_map))
+
     content_token_idx_to_word_idx_mapping =
       for {token, i} <- Enum.with_index(content_tokens), reduce: [] do
         acc ->
@@ -196,7 +197,7 @@ defmodule TFLiteElixir.MobileBert do
 
   defp excerpt_words(content_data, start_idx, end_idx) do
     pattern =
-      Enum.slice(content_data.content_words, start_idx..end_idx-1)
+      Enum.slice(content_data.content_words, start_idx..(end_idx - 1))
       |> Enum.map(&Regex.escape/1)
       |> Enum.join("\\s+")