fix: update inference main, tflite docs

nglehuy · nglehuy · commit c8d9f06fa700 · 2024-05-26T21:19:13.000+07:00
diff --git a/docs/tutorials/tflite.md b/docs/tutorials/tflite.md
@@ -11,7 +11,7 @@
 ## Conversion
 
 ```bash
-python3 examples/train.py \
+python3 examples/tflite.py \
     --config-path=/path/to/config.yml.j2 \
     --h5=/path/to/weight.h5 \
     --bs=1 \ # Batch size
diff --git a/examples/inferences/main.py b/examples/inferences/main.py
@@ -55,11 +55,8 @@ def main(
             previous_decoder_states=model.get_initial_decoder_states(),
         )
     )
-    print(outputs.tokens)
     transcript = tokenizer.detokenize(outputs.tokens)[0].numpy().decode("utf-8")
-
     logger.info(f"Transcript: {transcript}")
-    return transcript
 
 
 if __name__ == "__main__":

Original file line number	Diff line number	Diff line change
`@@ -55,11 +55,8 @@ def main(`
`55`	`55`	`previous_decoder_states=model.get_initial_decoder_states(),`
`56`	`56`	`)`
`57`	`57`	`)`
`58`		`- print(outputs.tokens)`
`59`	`58`	`transcript = tokenizer.detokenize(outputs.tokens)[0].numpy().decode("utf-8")`
`60`		`-`
`61`	`59`	`logger.info(f"Transcript: {transcript}")`
`62`		`- return transcript`
`63`	`60`
`64`	`61`
`65`	`62`	`if __name__ == "__main__":`