From d0f6c8c77af7b3d7012324b636ecafb8cd63e1c2 Mon Sep 17 00:00:00 2001
From: Joel Mathew Thomas <90510078+joelmathewthomas@users.noreply.github.com>
Date: Thu, 26 Dec 2024 16:30:19 +0530
Subject: [PATCH] disable cuda for classify.py

---
 src/preprocessing/classify.py | 5 +++++
 1 file changed, 5 insertions(+)

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')