Fixes
This commit is contained in:
@@ -2,11 +2,11 @@
|
||||
import sentry_sdk
|
||||
import os
|
||||
import requests
|
||||
import tempfile
|
||||
import numpy
|
||||
import json
|
||||
import uwsgi
|
||||
import pickle
|
||||
|
||||
from urllib.parse import urljoin
|
||||
from config import Config
|
||||
from pyAudioAnalysis.audioTrainTest import load_model, load_model_knn, classifier_wrapper
|
||||
from pyAudioAnalysis import audioBasicIO
|
||||
@@ -41,13 +41,13 @@ class ModelMemer:
|
||||
r.raise_for_status()
|
||||
self.model_details = r.json()
|
||||
|
||||
r = requests.get(self.model_details['files']['model'])
|
||||
r = requests.get(urljoin(Config.MODEL_INFO_URL,self.model_details['files']['model']))
|
||||
r.raise_for_status()
|
||||
|
||||
with open(model_file, 'wb') as f:
|
||||
f.write(r.content)
|
||||
|
||||
r = requests.get(self.model_details['files']['means'])
|
||||
r = requests.get(urljoin(Config.MODEL_INFO_URL,self.model_details['files']['means']))
|
||||
r.raise_for_status()
|
||||
|
||||
with open(means_file, 'wb') as f:
|
||||
@@ -109,7 +109,7 @@ def main():
|
||||
|
||||
while True:
|
||||
message = uwsgi.mule_get_msg()
|
||||
task = json.loads(message)
|
||||
task = pickle.loads(message)
|
||||
audio_file_path = task['audio_file_path']
|
||||
description = task['description']
|
||||
try:
|
||||
|
||||
Reference in New Issue
Block a user