diff --git a/model_service/views/cnn_view.py b/model_service/views/cnn_view.py index b1a608f..149130e 100644 --- a/model_service/views/cnn_view.py +++ b/model_service/views/cnn_view.py @@ -9,6 +9,8 @@ from utils import json_required, storage, ensure_buckets class CNNView(FlaskView): + route_base = 'cnn' + aimodel_schema = AIModelSchema(many=False) aimodels_schema = AIModelSchema(many=True, exclude=['timestamp', 'details']) default_schema = DefaultSchema(many=False) diff --git a/model_service/views/svm_view.py b/model_service/views/svm_view.py index 76124f0..765d531 100644 --- a/model_service/views/svm_view.py +++ b/model_service/views/svm_view.py @@ -12,6 +12,8 @@ from pyAudioAnalysis.audioTrainTest import load_model, load_model_knn class SVMView(FlaskView): + route_base = 'svm' + aimodel_schema = AIModelSchema(many=False) aimodels_schema = AIModelSchema(many=True, exclude=['timestamp', 'details']) default_schema = DefaultSchema(many=False)