diff --git a/src/py3.x/tensorflow2.x/config.py b/src/py3.x/tensorflow2.x/config.py
index 985fc173..8d149749 100644
--- a/src/py3.x/tensorflow2.x/config.py
+++ b/src/py3.x/tensorflow2.x/config.py
@@ -7,6 +7,7 @@
 
 class Config(object):
     poetry_file = 'poetry.txt'
+    weight_file = 'poetry_model.h5'
     data_file = 'EmotionData.xlsx'
     model_file = 'EmotionModel.h5'
     vocab_list = 'vocal_list.pkl'