Allow all models to appear always

TailUFPB · Jun 11, 2024 · dc159bd · dc159bd
1 parent fb60b0e
commit dc159bd
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 0 deletions.
diff --git 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
diff --git a/api/models/emotion_pipeline.pkl b/api/models/emotion_pipeline.pkl
diff --git a/api/models/hate_speech.pkl b/api/models/hate_speech.pkl
diff --git a/api/models/text_classification_pipeline.pkl b/api/models/text_classification_pipeline.pkl