diff --git a/dti_reviewer/backend/environment.yml b/dti_reviewer/backend/environment.yml index 7ca7893..0378e21 100644 --- a/dti_reviewer/backend/environment.yml +++ b/dti_reviewer/backend/environment.yml @@ -12,4 +12,4 @@ dependencies: - sphinx - celery - redis-py - + - pytables diff --git a/dti_reviewer/backend/tasks.py b/dti_reviewer/backend/tasks.py index a510af8..46155b7 100644 --- a/dti_reviewer/backend/tasks.py +++ b/dti_reviewer/backend/tasks.py @@ -1,5 +1,5 @@ from celery_app import celery -from similarity_engine import SimilarityEngine +from similarity_engine import SimilarityEngineOrcid from sklearn.metrics.pairwise import cosine_similarity engine = None @@ -7,7 +7,7 @@ def initialize_similarity_engine(): global engine if engine is None: - engine = SimilarityEngine() + engine = SimilarityEngineOrcid() engine.load_index_or_build()