add logger

run_demo.py

@@ -1,5 +1,14 @@
-from transformers import pipeline
+import logging
+
 import gradio as gr
+from transformers import pipeline
+
+logging.basicConfig(
+    format="%(asctime)s [%(levelname)s] [NLU] [Trainer] %(message)s",
+    datefmt="%Y-%m-%dT%H:%M:%SZ",
+)
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.DEBUG)
 
 pipe = pipeline(model="bhuang/wav2vec2-xls-r-1b-cv9-fr")
 
@@ -7,6 +16,7 @@ pipe = pipeline(model="bhuang/wav2vec2-xls-r-1b-cv9-fr")
 def transcribe(audio):
     # text = pipe(audio, chunk_length_s=30, stride_length_s=5)["text"]
     text = pipe(audio)["text"]
+    logger.info(f"Transcription for {audio}: {text}")
     return text
 
 
@@ -19,4 +29,5 @@ iface = gr.Interface(
     allow_flagging="never",
 )
 
+# iface.launch(server_name="0.0.0.0", debug=True, share=False)
 iface.launch()

run_demo_streaming.py

@@ -1,5 +1,14 @@
-from transformers import pipeline
+import logging
+
 import gradio as gr
+from transformers import pipeline
+
+logging.basicConfig(
+    format="%(asctime)s [%(levelname)s] [NLU] [Trainer] %(message)s",
+    datefmt="%Y-%m-%dT%H:%M:%SZ",
+)
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.INFO)
 
 pipe = pipeline(model="bhuang/wav2vec2-xls-r-1b-cv9-fr")
 
@@ -7,6 +16,7 @@ pipe = pipeline(model="bhuang/wav2vec2-xls-r-1b-cv9-fr")
 def transcribe(audio, state=""):
     text = pipe(audio, chunk_length_s=5, stride_length_s=1)["text"]
     state += text + " "
+    logger.info(f"Transcription for {audio}: {state}")
     return state, state