disabled cuda for yamnet model, and removed multi-env setup scripts, revert back to single env

This commit is contained in:
Joel Mathew Thomas
2025-02-24 20:49:16 +05:30
parent 90d3076e65
commit 10921cdad3
7 changed files with 32 additions and 221 deletions
+4
View File
@@ -5,6 +5,10 @@ import numpy as np
import csv
import os
# Force TensorFlow to use only CPU
tf.config.set_visible_devices([], 'GPU')
model = hub.load('https://tfhub.dev/google/yamnet/1')
#Find the name of the class with the top score when mean-aggregated across frames.