Add correct default get classifier case

TailUFPB · Jun 11, 2024 · 3de4129 · 3de4129
1 parent 57d2829
commit 3de4129
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/api/available_classifiers.py b/api/available_classifiers.py
@@ -5,11 +5,11 @@ def get_available_classifiers():
 
  # Verifica se o diretório 'models' existe
  if not model_folder:
- return {}
+ return {0 : "emotion_pipeline.pkl", 1 : "hate_speech.pkl", 2 : "text_classification_pipeline.pkl"}
 
  # Obtém a lista de arquivos no diretório 'models'
  model_files = os.listdir(model_folder)
- classifiers = {0 : "emotion_pipeline.pkl", 1 : "hate_speech.pkl", 2 : "text_classification_pipeline.pkl"}
+ classifiers = {}
 
  for file in model_files:
  classifiers[len(classifiers)] = file

diff --git a/package.json b/package.json
@@ -5,7 +5,7 @@
  "name": "TailLinguifAI",
  "email": ""
  },
- "version": "0.3.9",
+ "version": "0.3.5",
  "main": "./public/electron.js",
  "homepage": "./",
  "private": true,