This commit is contained in:
Saifeddine ALOUI 2023-08-03 01:41:59 +02:00
commit 4ed1be30dc

4
app.py
View File

@ -776,12 +776,16 @@ class LoLLMsWebUI(LoLLMsAPPI):
def list_personalities_categories(self): def list_personalities_categories(self):
language = request.args.get('language') language = request.args.get('language')
if language is None:
language = 'english'
personalities_categories_dir = self.lollms_paths.personalities_zoo_path/f'{language}' # replace with the actual path to the models folder personalities_categories_dir = self.lollms_paths.personalities_zoo_path/f'{language}' # replace with the actual path to the models folder
personalities_categories = [f.stem for f in personalities_categories_dir.iterdir() if f.is_dir() and not f.name.startswith(".")] personalities_categories = [f.stem for f in personalities_categories_dir.iterdir() if f.is_dir() and not f.name.startswith(".")]
return jsonify(personalities_categories) return jsonify(personalities_categories)
def list_personalities(self): def list_personalities(self):
language = request.args.get('language') language = request.args.get('language')
if language is None:
language = 'english'
category = request.args.get('category') category = request.args.get('category')
if not category: if not category:
return jsonify([]) return jsonify([])