From 7295d4ff3d0b5b2bb6241a2e3df217ffa828b135 Mon Sep 17 00:00:00 2001
From: Joel Mathew Thomas <90510078+joelmathewthomas@users.noreply.github.com>
Date: Sat, 25 Jan 2025 17:38:27 +0530
Subject: [PATCH] enable cuda for classify_audio and demucs_wrapper

---
 src/preprocessing/classify.py | 4 ----
 1 file changed, 4 deletions(-)

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.