diff --git a/src/preprocessing/classify.py b/src/preprocessing/classify.py index 05363dc..f835eea 100644 --- a/src/preprocessing/classify.py +++ b/src/preprocessing/classify.py @@ -5,10 +5,6 @@ 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') #Find the name of the class with the top score when mean-aggregated across frames.