disable CUDA stub cache download and GPU discovery

app.py

@@ -1,5 +1,8 @@
 import streamlit as st
 import tensorflow as tf
+
+# avoid massive CUDA stub cache downloads
+tf.config.set_visible_devices([], 'GPU')
 import numpy as np
 import librosa
 import matplotlib.pyplot as plt
@@ -7,6 +10,17 @@ import librosa.display
 import tempfile
 import os
 
+os.environ["HF_HOME"] = "/data/huggingface"
+os.environ["HF_HUB_CACHE"] = "/data/huggingface/hub"
+os.environ["HF_DATASETS_CACHE"] = "/data/huggingface/datasets"
+os.environ["TRANSFORMERS_CACHE"] = "/data/huggingface/transformers"
+os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
+os.environ["XDG_CACHE_HOME"] = "/data/cache"
+
+# Disable tensordlow GPU discovery
+os.environ["CUDA_VISIBLE_DEVICES"] = "-1" 
+
+
 # Load the trained model #test
 @st.cache_resource
 def load_model():