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

added lapatoloas

This commit is contained in:
Pünkösd Marcell 2021-12-02 21:43:58 +01:00
parent 6c259c1451
commit 99aa33b1df
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
svm_prefilter_service/mule.py
View File

@ -5,6 +5,8 @@ import requests
import json import json
import uwsgi import uwsgi
import pickle import pickle
from threading import Thread
from queue import Queue
from urllib.parse import urljoin from urllib.parse import urljoin
from config import Config from config import Config
@ -111,11 +113,20 @@ def run_classification(audio_file_path: str, memer: ModelMemer):
return bool((class_id == memer.target_id) and (probability[class_id] > 0.5)) return bool((class_id == memer.target_id) and (probability[class_id] > 0.5))
def main(): def lapatolas(q: Queue):
memer = ModelMemer()
while True: while True:
message = uwsgi.mule_get_msg() message = uwsgi.mule_get_msg()
q.put(message)
def main():
memer = ModelMemer()
requeue = Queue()
Thread(target=lapatolas, args=(requeue,), daemon=True).start()
while True:
message = requeue.get(block=True)
task = pickle.loads(message) task = pickle.loads(message)
audio_file_path = task['audio_file_path'] audio_file_path = task['audio_file_path']
description = task['description'] description = task['description']