diff --git a/install_model.py b/install_model.py index 1400cf93f..d9395bc5d 100644 --- a/install_model.py +++ b/install_model.py @@ -12,8 +12,7 @@ import os.path from transformers import GPT2TokenizerFast import sentence_transformers -prefix_dir = os.path.join(os.path.dirname(os.path.abspath(os.getcwd())), 'model') - +prefix_dir = "/opt/maxkb/model" model_config = [ { 'download_params': { diff --git a/main.py b/main.py index 8f8802302..1a702706f 100644 --- a/main.py +++ b/main.py @@ -44,11 +44,11 @@ def perform_db_migrate(): def start_services(): management.call_command('migrate') - management.call_command('runserver',"0.0.0.0:8000") + management.call_command('runserver', "0.0.0.0:8000") if __name__ == '__main__': - os.environ['TRANSFORMERS_CACHE'] = '/opt/maxkb/model' + os.environ['TRANSFORMERS_CACHE'] = '/opt/maxkb/model/base' parser = argparse.ArgumentParser( description=""" qabot service control tools;