diff --git a/api/available_classifiers.py b/api/available_classifiers.py index 76bf7aea..da0ad0cc 100644 --- a/api/available_classifiers.py +++ b/api/available_classifiers.py @@ -13,5 +13,8 @@ def get_available_classifiers(): for file in model_files: classifiers[len(classifiers)] = file + classifiers[len(classifiers)] = "emotion_pipeline.pkl" + classifiers[len(classifiers)] = "hate_speech.pkl" + classifiers[len(classifiers)] = "text_classification_pipeline.pkl" return classifiers \ No newline at end of file diff --git a/api/models/emotion_pipeline.pkl b/api/models/emotion_pipeline.pkl deleted file mode 100644 index 37bd3550..00000000 Binary files a/api/models/emotion_pipeline.pkl and /dev/null differ diff --git a/api/models/hate_speech.pkl b/api/models/hate_speech.pkl deleted file mode 100644 index edde8a41..00000000 Binary files a/api/models/hate_speech.pkl and /dev/null differ diff --git a/api/models/text_classification_pipeline.pkl b/api/models/text_classification_pipeline.pkl deleted file mode 100644 index 893cf9ec..00000000 Binary files a/api/models/text_classification_pipeline.pkl and /dev/null differ