czyoung commited on
Commit
1505dd9
·
verified ·
1 Parent(s): 24dc2fc

Fix numpy scalar issue

Browse files
Files changed (1) hide show
  1. sonogram.py +1 -1
sonogram.py CHANGED
@@ -61,7 +61,7 @@ class Sonogram():
61
  self.groupClassifier = pickle.load(f)
62
 
63
  def classifyEmbedding(self,embedding):
64
- return int(self.groupClassifier.predict(embedding.reshape(1, -1)))
65
 
66
  def processFile(self,filePath):
67
  # Loading audio file
 
61
  self.groupClassifier = pickle.load(f)
62
 
63
  def classifyEmbedding(self,embedding):
64
+ return int(self.groupClassifier.predict(embedding.reshape(1, -1)).item())
65
 
66
  def processFile(self,filePath):
67
  # Loading audio file