enable cuda for classify_audio and demucs_wrapper

2025-01-25 17:38:27 +05:30
parent e0c19002d1
commit 7295d4ff3d
1 changed files with 0 additions and 4 deletions
@@ -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.