diff --git a/.idea/.gitignore b/.idea/.gitignore
new file mode 100644
index 0000000..26d3352
--- /dev/null
+++ b/.idea/.gitignore
@@ -0,0 +1,3 @@
+# Default ignored files
+/shelf/
+/workspace.xml
diff --git a/.idea/QueryCorrection.iml b/.idea/QueryCorrection.iml
new file mode 100644
index 0000000..fa63b11
--- /dev/null
+++ b/.idea/QueryCorrection.iml
@@ -0,0 +1,11 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$" />
+    <orderEntry type="jdk" jdkName="Python 3.6 (tensorflow2.1)" jdkType="Python SDK" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+  <component name="TestRunnerService">
+    <option name="PROJECT_TEST_RUNNER" value="pytest" />
+  </component>
+</module>
\ No newline at end of file
diff --git a/.idea/inspectionProfiles/Project_Default.xml b/.idea/inspectionProfiles/Project_Default.xml
new file mode 100644
index 0000000..c47c073
--- /dev/null
+++ b/.idea/inspectionProfiles/Project_Default.xml
@@ -0,0 +1,221 @@
+<component name="InspectionProjectProfileManager">
+  <profile version="1.0">
+    <option name="myName" value="Project Default" />
+    <inspection_tool class="PyPackageRequirementsInspection" enabled="true" level="WARNING" enabled_by_default="true">
+      <option name="ignoredPackages">
+        <value>
+          <list size="185">
+            <item index="0" class="java.lang.String" itemvalue="tensorflow" />
+            <item index="1" class="java.lang.String" itemvalue="sklearn" />
+            <item index="2" class="java.lang.String" itemvalue="pandas" />
+            <item index="3" class="java.lang.String" itemvalue="matplotlib" />
+            <item index="4" class="java.lang.String" itemvalue="numpy" />
+            <item index="5" class="java.lang.String" itemvalue="flask_json" />
+            <item index="6" class="java.lang.String" itemvalue="flask_cors" />
+            <item index="7" class="java.lang.String" itemvalue="flask" />
+            <item index="8" class="java.lang.String" itemvalue="GPUtil" />
+            <item index="9" class="java.lang.String" itemvalue="flask_compress" />
+            <item index="10" class="java.lang.String" itemvalue="absl-py" />
+            <item index="11" class="java.lang.String" itemvalue="protobuf" />
+            <item index="12" class="java.lang.String" itemvalue="scipy" />
+            <item index="13" class="java.lang.String" itemvalue="tensorflow-estimator" />
+            <item index="14" class="java.lang.String" itemvalue="six" />
+            <item index="15" class="java.lang.String" itemvalue="threadpoolctl" />
+            <item index="16" class="java.lang.String" itemvalue="Werkzeug" />
+            <item index="17" class="java.lang.String" itemvalue="scikit-learn" />
+            <item index="18" class="java.lang.String" itemvalue="h5py" />
+            <item index="19" class="java.lang.String" itemvalue="typing-extensions" />
+            <item index="20" class="java.lang.String" itemvalue="gast" />
+            <item index="21" class="java.lang.String" itemvalue="importlib-metadata" />
+            <item index="22" class="java.lang.String" itemvalue="tensorboard" />
+            <item index="23" class="java.lang.String" itemvalue="certifi" />
+            <item index="24" class="java.lang.String" itemvalue="cached-property" />
+            <item index="25" class="java.lang.String" itemvalue="grpcio" />
+            <item index="26" class="java.lang.String" itemvalue="bert4keras" />
+            <item index="27" class="java.lang.String" itemvalue="Flask" />
+            <item index="28" class="java.lang.String" itemvalue="keras-bert" />
+            <item index="29" class="java.lang.String" itemvalue="seqeval" />
+            <item index="30" class="java.lang.String" itemvalue="Keras" />
+            <item index="31" class="java.lang.String" itemvalue="tensorflow_gpu" />
+            <item index="32" class="java.lang.String" itemvalue="scikit_learn" />
+            <item index="33" class="java.lang.String" itemvalue="tokenizers" />
+            <item index="34" class="java.lang.String" itemvalue="transformers" />
+            <item index="35" class="java.lang.String" itemvalue="pypi-kenlm" />
+            <item index="36" class="java.lang.String" itemvalue="sacremoses" />
+            <item index="37" class="java.lang.String" itemvalue="joblib" />
+            <item index="38" class="java.lang.String" itemvalue="huggingface-hub" />
+            <item index="39" class="java.lang.String" itemvalue="opencv-python" />
+            <item index="40" class="java.lang.String" itemvalue="PyYAML" />
+            <item index="41" class="java.lang.String" itemvalue="xpinyin" />
+            <item index="42" class="java.lang.String" itemvalue="wrapt" />
+            <item index="43" class="java.lang.String" itemvalue="text-utils" />
+            <item index="44" class="java.lang.String" itemvalue="cycler" />
+            <item index="45" class="java.lang.String" itemvalue="click" />
+            <item index="46" class="java.lang.String" itemvalue="Jinja2" />
+            <item index="47" class="java.lang.String" itemvalue="pycorrector" />
+            <item index="48" class="java.lang.String" itemvalue="regex" />
+            <item index="49" class="java.lang.String" itemvalue="filelock" />
+            <item index="50" class="java.lang.String" itemvalue="zipp" />
+            <item index="51" class="java.lang.String" itemvalue="pytz" />
+            <item index="52" class="java.lang.String" itemvalue="xlrd" />
+            <item index="53" class="java.lang.String" itemvalue="notebook" />
+            <item index="54" class="java.lang.String" itemvalue="Pillow" />
+            <item index="55" class="java.lang.String" itemvalue="tensorflow-gpu" />
+            <item index="56" class="java.lang.String" itemvalue="greenlet" />
+            <item index="57" class="java.lang.String" itemvalue="google-auth-oauthlib" />
+            <item index="58" class="java.lang.String" itemvalue="cffi" />
+            <item index="59" class="java.lang.String" itemvalue="tensorboard-data-server" />
+            <item index="60" class="java.lang.String" itemvalue="asgiref" />
+            <item index="61" class="java.lang.String" itemvalue="tf-slim" />
+            <item index="62" class="java.lang.String" itemvalue="grpcio-tools" />
+            <item index="63" class="java.lang.String" itemvalue="MarkupSafe" />
+            <item index="64" class="java.lang.String" itemvalue="bitarray" />
+            <item index="65" class="java.lang.String" itemvalue="attrs" />
+            <item index="66" class="java.lang.String" itemvalue="bleach" />
+            <item index="67" class="java.lang.String" itemvalue="cachetools" />
+            <item index="68" class="java.lang.String" itemvalue="lshash3" />
+            <item index="69" class="java.lang.String" itemvalue="itsdangerous" />
+            <item index="70" class="java.lang.String" itemvalue="webencodings" />
+            <item index="71" class="java.lang.String" itemvalue="google-auth" />
+            <item index="72" class="java.lang.String" itemvalue="gensim" />
+            <item index="73" class="java.lang.String" itemvalue="testpath" />
+            <item index="74" class="java.lang.String" itemvalue="pickleshare" />
+            <item index="75" class="java.lang.String" itemvalue="defusedxml" />
+            <item index="76" class="java.lang.String" itemvalue="PyMsgBox" />
+            <item index="77" class="java.lang.String" itemvalue="pycparser" />
+            <item index="78" class="java.lang.String" itemvalue="torchvision" />
+            <item index="79" class="java.lang.String" itemvalue="redis" />
+            <item index="80" class="java.lang.String" itemvalue="pyasn1-modules" />
+            <item index="81" class="java.lang.String" itemvalue="ipython-genutils" />
+            <item index="82" class="java.lang.String" itemvalue="Pygments" />
+            <item index="83" class="java.lang.String" itemvalue="astunparse" />
+            <item index="84" class="java.lang.String" itemvalue="english" />
+            <item index="85" class="java.lang.String" itemvalue="monotonic" />
+            <item index="86" class="java.lang.String" itemvalue="jsonschema" />
+            <item index="87" class="java.lang.String" itemvalue="pywin32" />
+            <item index="88" class="java.lang.String" itemvalue="qtconsole" />
+            <item index="89" class="java.lang.String" itemvalue="terminado" />
+            <item index="90" class="java.lang.String" itemvalue="pyperclip" />
+            <item index="91" class="java.lang.String" itemvalue="jupyter-client" />
+            <item index="92" class="java.lang.String" itemvalue="jupyterlab-pygments" />
+            <item index="93" class="java.lang.String" itemvalue="ipykernel" />
+            <item index="94" class="java.lang.String" itemvalue="nbconvert" />
+            <item index="95" class="java.lang.String" itemvalue="jedi" />
+            <item index="96" class="java.lang.String" itemvalue="flatbuffers" />
+            <item index="97" class="java.lang.String" itemvalue="idna" />
+            <item index="98" class="java.lang.String" itemvalue="rsa" />
+            <item index="99" class="java.lang.String" itemvalue="decorator" />
+            <item index="100" class="java.lang.String" itemvalue="pandocfilters" />
+            <item index="101" class="java.lang.String" itemvalue="py4j" />
+            <item index="102" class="java.lang.String" itemvalue="pyasn1" />
+            <item index="103" class="java.lang.String" itemvalue="requests" />
+            <item index="104" class="java.lang.String" itemvalue="websocket-client" />
+            <item index="105" class="java.lang.String" itemvalue="pyrsistent" />
+            <item index="106" class="java.lang.String" itemvalue="jupyter" />
+            <item index="107" class="java.lang.String" itemvalue="tensorboard-plugin-wit" />
+            <item index="108" class="java.lang.String" itemvalue="nest-asyncio" />
+            <item index="109" class="java.lang.String" itemvalue="smart-open" />
+            <item index="110" class="java.lang.String" itemvalue="prompt-toolkit" />
+            <item index="111" class="java.lang.String" itemvalue="ipywidgets" />
+            <item index="112" class="java.lang.String" itemvalue="PyScreeze" />
+            <item index="113" class="java.lang.String" itemvalue="tornado" />
+            <item index="114" class="java.lang.String" itemvalue="Send2Trash" />
+            <item index="115" class="java.lang.String" itemvalue="torch" />
+            <item index="116" class="java.lang.String" itemvalue="et-xmlfile" />
+            <item index="117" class="java.lang.String" itemvalue="toml" />
+            <item index="118" class="java.lang.String" itemvalue="mistune" />
+            <item index="119" class="java.lang.String" itemvalue="Django" />
+            <item index="120" class="java.lang.String" itemvalue="jupyter-console" />
+            <item index="121" class="java.lang.String" itemvalue="argon2-cffi" />
+            <item index="122" class="java.lang.String" itemvalue="PyGetWindow" />
+            <item index="123" class="java.lang.String" itemvalue="ujson" />
+            <item index="124" class="java.lang.String" itemvalue="traitlets" />
+            <item index="125" class="java.lang.String" itemvalue="pywinpty" />
+            <item index="126" class="java.lang.String" itemvalue="nltk" />
+            <item index="127" class="java.lang.String" itemvalue="nbclient" />
+            <item index="128" class="java.lang.String" itemvalue="QtPy" />
+            <item index="129" class="java.lang.String" itemvalue="jupyterlab-widgets" />
+            <item index="130" class="java.lang.String" itemvalue="pyzmq" />
+            <item index="131" class="java.lang.String" itemvalue="sentencepiece" />
+            <item index="132" class="java.lang.String" itemvalue="oauthlib" />
+            <item index="133" class="java.lang.String" itemvalue="entrypoints" />
+            <item index="134" class="java.lang.String" itemvalue="pypinyin" />
+            <item index="135" class="java.lang.String" itemvalue="cryptography" />
+            <item index="136" class="java.lang.String" itemvalue="zope.interface" />
+            <item index="137" class="java.lang.String" itemvalue="PyTweening" />
+            <item index="138" class="java.lang.String" itemvalue="MouseInfo" />
+            <item index="139" class="java.lang.String" itemvalue="backcall" />
+            <item index="140" class="java.lang.String" itemvalue="zhon" />
+            <item index="141" class="java.lang.String" itemvalue="widgetsnbextension" />
+            <item index="142" class="java.lang.String" itemvalue="deepctr" />
+            <item index="143" class="java.lang.String" itemvalue="async-generator" />
+            <item index="144" class="java.lang.String" itemvalue="supervisor" />
+            <item index="145" class="java.lang.String" itemvalue="jupyter-core" />
+            <item index="146" class="java.lang.String" itemvalue="jieba" />
+            <item index="147" class="java.lang.String" itemvalue="python-Levenshtein" />
+            <item index="148" class="java.lang.String" itemvalue="wcwidth" />
+            <item index="149" class="java.lang.String" itemvalue="typeguard" />
+            <item index="150" class="java.lang.String" itemvalue="docker" />
+            <item index="151" class="java.lang.String" itemvalue="sqlparse" />
+            <item index="152" class="java.lang.String" itemvalue="bert-tensorflow" />
+            <item index="153" class="java.lang.String" itemvalue="requests-oauthlib" />
+            <item index="154" class="java.lang.String" itemvalue="urllib3" />
+            <item index="155" class="java.lang.String" itemvalue="zope.event" />
+            <item index="156" class="java.lang.String" itemvalue="Cython" />
+            <item index="157" class="java.lang.String" itemvalue="parso" />
+            <item index="158" class="java.lang.String" itemvalue="nbformat" />
+            <item index="159" class="java.lang.String" itemvalue="ipython" />
+            <item index="160" class="java.lang.String" itemvalue="sentence-transformers" />
+            <item index="161" class="java.lang.String" itemvalue="packaging" />
+            <item index="162" class="java.lang.String" itemvalue="pansi" />
+            <item index="163" class="java.lang.String" itemvalue="prometheus-client" />
+            <item index="164" class="java.lang.String" itemvalue="chardet" />
+            <item index="165" class="java.lang.String" itemvalue="tqdm" />
+            <item index="166" class="java.lang.String" itemvalue="neotime" />
+            <item index="167" class="java.lang.String" itemvalue="PyRect" />
+            <item index="168" class="java.lang.String" itemvalue="colorama" />
+            <item index="169" class="java.lang.String" itemvalue="python-dateutil" />
+            <item index="170" class="java.lang.String" itemvalue="kiwisolver" />
+            <item index="171" class="java.lang.String" itemvalue="pyparsing" />
+            <item index="172" class="java.lang.String" itemvalue="Markdown" />
+            <item index="173" class="java.lang.String" itemvalue="beautifulsoup4" />
+            <item index="174" class="java.lang.String" itemvalue="keras-transformer" />
+            <item index="175" class="java.lang.String" itemvalue="future" />
+            <item index="176" class="java.lang.String" itemvalue="keras-embed-sim" />
+            <item index="177" class="java.lang.String" itemvalue="charset-normalizer" />
+            <item index="178" class="java.lang.String" itemvalue="soupsieve" />
+            <item index="179" class="java.lang.String" itemvalue="keras-position-wise-feed-forward" />
+            <item index="180" class="java.lang.String" itemvalue="keras-pos-embd" />
+            <item index="181" class="java.lang.String" itemvalue="keras-self-attention" />
+            <item index="182" class="java.lang.String" itemvalue="keras-layer-normalization" />
+            <item index="183" class="java.lang.String" itemvalue="keras-multi-head" />
+            <item index="184" class="java.lang.String" itemvalue="pydot" />
+          </list>
+        </value>
+      </option>
+    </inspection_tool>
+    <inspection_tool class="PyPep8Inspection" enabled="true" level="WEAK WARNING" enabled_by_default="true">
+      <option name="ignoredErrors">
+        <list>
+          <option value="E501" />
+          <option value="W605" />
+        </list>
+      </option>
+    </inspection_tool>
+    <inspection_tool class="PyPep8NamingInspection" enabled="true" level="WEAK WARNING" enabled_by_default="true">
+      <option name="ignoredErrors">
+        <list>
+          <option value="N802" />
+        </list>
+      </option>
+    </inspection_tool>
+    <inspection_tool class="PyUnresolvedReferencesInspection" enabled="true" level="WARNING" enabled_by_default="true">
+      <option name="ignoredIdentifiers">
+        <list>
+          <option value="requests.models.Response.get" />
+          <option value="list.*" />
+        </list>
+      </option>
+    </inspection_tool>
+  </profile>
+</component>
\ No newline at end of file
diff --git a/.idea/inspectionProfiles/profiles_settings.xml b/.idea/inspectionProfiles/profiles_settings.xml
new file mode 100644
index 0000000..105ce2d
--- /dev/null
+++ b/.idea/inspectionProfiles/profiles_settings.xml
@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>
\ No newline at end of file
diff --git a/.idea/misc.xml b/.idea/misc.xml
new file mode 100644
index 0000000..e2a8e0c
--- /dev/null
+++ b/.idea/misc.xml
@@ -0,0 +1,4 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.6 (tensorflow2.1)" project-jdk-type="Python SDK" />
+</project>
\ No newline at end of file
diff --git a/.idea/modules.xml b/.idea/modules.xml
new file mode 100644
index 0000000..4ee2871
--- /dev/null
+++ b/.idea/modules.xml
@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/QueryCorrection.iml" filepath="$PROJECT_DIR$/.idea/QueryCorrection.iml" />
+    </modules>
+  </component>
+</project>
\ No newline at end of file
diff --git a/.idea/vcs.xml b/.idea/vcs.xml
new file mode 100644
index 0000000..94a25f7
--- /dev/null
+++ b/.idea/vcs.xml
@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="$PROJECT_DIR$" vcs="Git" />
+  </component>
+</project>
\ No newline at end of file
diff --git a/demo.py b/demo.py
index e0e6e5f..6c23b27 100644
--- a/demo.py
+++ b/demo.py
@@ -57,7 +57,7 @@ def build_model():
 
 def test():
     corrector = WordCorrect()
-    word = '我门'
+    word = '窝门'
     word_pinyin = ','.join(lazy_pinyin(word))
     candiwords = corrector.edit1(word)
     print(candiwords)