fix: app.py

app.py

@@ -1,8 +1,15 @@
-import torch
-import numpy as np
 import gradio as gr
-from transformers import pipeline
+import numpy as np
+import torch
+import transformers
 from pathlib import Path
+from transformers import pipeline
+from transformers.utils import logging
+
+# Log
+
+#logging.set_verbosity_debug()
+logger = logging.get_logger("transformers")
 
 # Pipelines
 
@@ -44,6 +51,8 @@ def transcribe(audio: str | Path | bytes | tuple[int, np.ndarray] | None):
         return "..."
     # TODO Manage str/Path
 
+    logger.debug("Transcribe")
+
     text = ""
     # https://huggingface.co/docs/transformers/main_classes/pipelines#transformers.AutomaticSpeechRecognitionPipeline.__call__
     # Whisper input format for tuple differ from output provided by gradio audio component
@@ -52,7 +61,12 @@ def transcribe(audio: str | Path | bytes | tuple[int, np.ndarray] | None):
         transcript = asr(inputs)
         text = transcript['text']
 
+    logger.debug("Tokenize:[" + text + "]")
+
     entities = tc(text)
+
+    #logger.debug("Classify:[" + entities + "]")
+
     # TODO Add Text Classification for sentiment analysis
     return {"text": text, "entities": entities}