diff --git a/api/available_classifiers.py b/api/available_classifiers.py index 9901ec50..76bf7aea 100644 --- 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 index ef56abb3..3cd47c80 100644 --- 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,