优化代码

Author: charlesXu86

Chatbot_Retrieval/settings.py

@@ -15,7 +15,6 @@
 # Build paths inside the project like this: os.path.join(BASE_DIR, ...)
 BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 
-
 # Quick-start development settings - unsuitable for production
 # See https://docs.djangoproject.com/en/2.2/howto/deployment/checklist/
 
@@ -27,7 +26,6 @@
 
 ALLOWED_HOSTS = ['*']
 
-
 # Application definition
 
 INSTALLED_APPS = [
@@ -37,8 +35,10 @@
     'django.contrib.sessions',
     'django.contrib.messages',
     'django.contrib.staticfiles',
+    'Chatbot_Retrieval_model'
 ]
 
+
 MIDDLEWARE = [
     'django.middleware.security.SecurityMiddleware',
     'django.contrib.sessions.middleware.SessionMiddleware',
@@ -70,17 +70,9 @@
 
 WSGI_APPLICATION = 'Chatbot_Retrieval.wsgi.application'
 
-
 # Database
 # https://docs.djangoproject.com/en/2.2/ref/settings/#databases
 
-DATABASES = {
-    'default': {
-        'ENGINE': 'django.db.backends.sqlite3',
-        'NAME': os.path.join(BASE_DIR, 'db.sqlite3'),
-    }
-}
-
 
 # Password validation
 # https://docs.djangoproject.com/en/2.2/ref/settings/#auth-password-validators
@@ -100,7 +92,6 @@
     },
 ]
 
-
 # Internationalization
 # https://docs.djangoproject.com/en/2.2/topics/i18n/
 
@@ -114,7 +105,6 @@
 
 USE_TZ = True
 
-
 # Static files (CSS, JavaScript, Images)
 # https://docs.djangoproject.com/en/2.2/howto/static-files/
 

Chatbot_Retrieval/urls.py

@@ -16,9 +16,11 @@
 from django.contrib import admin
 from django.urls import path, include
 
+
 urlpatterns = [
     path('admin/', admin.site.urls),
 
 
     path('api/', include("Chatbot_Retrival_rest.urls")),
+
 ]

Chatbot_Retrieval_model/Bert_sim/config_bert.py

@@ -23,7 +23,7 @@ def __init__(self):
         self.bert_config_file = '/Data/public/pretrained_models/tensorflow1.x/chinese-bert-wwm-ext/bert_config.json'
         self.vocab_file = '/Data/public/pretrained_models/tensorflow1.x/chinese-bert-wwm-ext/vocab.txt'
         self.data_dir = os.path.join(basedir, 'data/bert_sim/')
-        self.output_dir = basedir + '/Chatbot_Retrieval_model/Bert_sim/saved_model_bert'
+        self.output_dir = basedir + '/Chatbot_Retrieval_model_sss/Bert_sim/saved_model_bert'
         self.init_checkpoint = '/Data/public/pretrained_models/tensorflow1.x/chinese-bert-wwm-ext/bert_model.ckpt'
 
         self.do_lower_case = True