Merge pull request #284 from NexaAI/david/bugfix

suppress log print in streamlit
NexaAI · Nov 25, 2024 · bfe4fcd · bfe4fcd
2 parents 235b07e + d3c7373
commit bfe4fcd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/nexa/gguf/streamlit/streamlit_audio_lm.py b/nexa/gguf/streamlit/streamlit_audio_lm.py
@@ -7,6 +7,7 @@
 from st_audiorec import st_audiorec
 
 from nexa.general import pull_model
+from nexa.gguf.llama._utils_transformers import suppress_stdout_stderr
 from nexa.gguf.nexa_inference_audio_lm import NexaAudioLMInference
 
 # Initialize session state
@@ -45,7 +46,8 @@ def process_audio(nexa_model, audio_file, prompt=""):
 
     try:
         # Use the model's inference method directly
-        response = nexa_model.inference(temp_audio_path, prompt)
+        with suppress_stdout_stderr():
+            response = nexa_model.inference(temp_audio_path, prompt)
         return response
 
     except Exception as e: