diff --git a/src/preprocessing/classify.py b/src/preprocessing/classify.py index 59eadd6..05363dc 100644 --- a/src/preprocessing/classify.py +++ b/src/preprocessing/classify.py @@ -3,6 +3,11 @@ import tensorflow_hub as hub import librosa import numpy as np import csv +import os + +# Disable CUDA +os.environ["CUDA_VISIBLE_DEVICES"] = "-1" + model = hub.load('https://tfhub.dev/google/yamnet/1')