upgraded model

This commit is contained in:
ParisNeo 2023-05-05 22:35:08 +02:00
parent b2f5bd8625
commit 371a91595c
3 changed files with 7 additions and 3 deletions

1
.gitignore vendored
View File

@ -151,7 +151,6 @@ configs/*
# personalities other than the default one
personalities/*
!personalities/english/general/gpt4all_chatbot/config.yaml
# personalities other than the default one
databases/*

8
app.py
View File

@ -309,8 +309,12 @@ class Gpt4AllWebUI(GPT4AllAPI):
return jsonify(personalities_categories)
def list_personalities(self):
personalities_dir = Path(f'./personalities/{self.config["personality_language"]}/{self.config["personality_category"]}') # replace with the actual path to the models folder
personalities = [f.stem for f in personalities_dir.iterdir() if f.is_dir()]
try:
personalities_dir = Path(f'./personalities/{self.config["personality_language"]}/{self.config["personality_category"]}') # replace with the actual path to the models folder
personalities = [f.stem for f in personalities_dir.iterdir() if f.is_dir()]
except:
personalities=[]
print("nope")
return jsonify(personalities)
def list_languages(self):

View File

@ -6,6 +6,7 @@ pyyaml
markdown
pyllamacpp==2.1.1
gpt4all-j==0.2.1
pygptj
--find-links https://download.pytorch.org/whl/cu117
torch==2.0.0
torchvision