birbnetes/svm-prefilter-service
birbnetes/svm-prefilter-service
4
0
Fork 0
Code Pull Requests Releases Activity

Fixed model not downloading
All checks were successful
continuous-integration/drone/push Build is passing
Details

Browse Source
This commit is contained in:
Pünkösd Marcell 2021-12-18 03:58:28 +01:00
parent a23971a47a
commit b89038dc67
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
svm_prefilter_service/mule.py
View File

@ -36,7 +36,7 @@ def json_datetime_dumper(o):
class ModelMemer:
def __init__(self):
self._loaded_model = None
self._loaded_model = False
def download_model_if_needed(self):
models_dir = "/tmp/svm_model"
@ -79,6 +79,8 @@ class ModelMemer:
target_class_name = self.model_details['target_class_name']
self.target_id = self.classes.index(target_class_name)
self._loaded_model = True
def run_classification(audio_file_path: str, memer: ModelMemer):
memer.download_model_if_needed()