From 99aa33b1df64757930b9debe0b80f04b5fbf9b0d Mon Sep 17 00:00:00 2001 From: marcsello Date: Thu, 2 Dec 2021 21:43:58 +0100 Subject: [PATCH] added lapatoloas --- svm_prefilter_service/mule.py | 17 ++++++++++++++--- 1 file changed, 14 insertions(+), 3 deletions(-) diff --git a/svm_prefilter_service/mule.py b/svm_prefilter_service/mule.py index 5220212..7c634c4 100644 --- a/svm_prefilter_service/mule.py +++ b/svm_prefilter_service/mule.py @@ -5,6 +5,8 @@ import requests import json import uwsgi import pickle +from threading import Thread +from queue import Queue from urllib.parse import urljoin 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)) -def main(): - memer = ModelMemer() - +def lapatolas(q: Queue): while True: 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) audio_file_path = task['audio_file_path'] description = task['description']