Spaces:
Sleeping
Sleeping
Fix numpy scalar issue
Browse files- 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
|