diff --git a/handler.py b/handler.py
index 527091a..853957e 100644
--- a/handler.py
+++ b/handler.py
@@ -9,12 +9,26 @@
     """
 
 #     param1 = conf['param1']  # value_type: str # description: some description
-    start_words = conf['start_words']    #诗歌开始
-    prefix_words = conf['prefix_words']  #诗歌语境
-    max_gen_len = conf['max_gen_len']    #诗歌最大长度
-    # add your code
+    # start_words = conf['start_words']    #诗歌开始
+    # prefix_words = conf['prefix_words']  #诗歌语境
+    # max_gen_len = conf['max_gen_len']    #诗歌最大长度
+    # # add your code
+    # if __name__ == '__main__':
+    #     cnf = {start_words,prefix_words,max_gen_len}
+    #     result = main.gen(**cnf)
+    #     result = ''.json(result)
+    # return {'result': result}
+
     if __name__ == '__main__':
-        result = main.gen(start_words,prefix_words,max_gen_len)
-        result = ''.json(result)
-    return {'result': result}
+        start_words = conf['start_words']    #诗歌开始
+        prefix_words = conf['prefix_words']  #诗歌语境
+        max_gen_len = conf['max_gen_len']    #诗歌最大长度
+        cof={
+        "max_gen_len": max_gen_len,  # 生成诗歌最长长度
+        "prefix_words":prefix_words,  # 不是诗歌的组成部分，用来控制生成诗歌的意境
+        "start_words" : start_words  # 诗歌开始
+        }
+        result = main.gen(**cof)
+        result = ''.join(result)
+        return {'ret1':result}
     diff --git a/main.py b/main.py
index 443632b..8389137 100644
--- a/main.py
+++ b/main.py
@@ -33,31 +33,7 @@
     acrostic = False  # 是否是藏头诗
     model_prefix = 'checkpoints/tang'  # 模型保存路径
 
-# conf={
-#     "data_path":'tang.npz',  # 诗歌的文本文件存放路径
-#     "pickle_path":'tang.npz' , # 预处理好的二进制文件
-#     "author":None,  # 只学习某位作者的诗歌
-#     "constrain": None,  # 长度限制
-#     "category" :'poet.tang',  # 类别，唐诗还是宋诗歌(poet.song)
-#     "lr" :1e-3,
-#     "weight_decay": 1e-4,
-#     "use_gpu" : False,
-#     "epoch" : 20,
-#     "batch_size" : 128,
-#     "maxlen" : 125 , # 超过这个长度的之后字被丢弃，小于这个长度的在前面补空格
-#     "plot_every" : 20,  # 每20个batch 可视化一次
-#     "use_env" : True, # 是否使用visodm
-#     "env" : 'poetry' , # visdom env
-#     "max_gen_len" : 200 , # 生成诗歌最长长度
-#     "debug_file" : '/tmp/debugp',
-#     "model_path" : 'checkpoints/tang_199.pth',  # 预训练模型路径
-#     "prefix_words": '笑语盈盈暗香去' , # 不是诗歌的组成部分，用来控制生成诗歌的意境
-#     "start_words" : '语' , # 诗歌开始
-#     "acrostic" : False , # 是否是藏头诗
-#     "model_prefix" : 'checkpoints/tang'  # 模型保存路径
-# }
 opt = Config()
-# opt = conf
 
 
 def generate(model, start_words, ix2word, word2ix, prefix_words=None):
@@ -248,16 +224,19 @@
     gen_poetry = gen_acrostic if opt.acrostic else generate
     result = gen_poetry(model, start_words, ix2word, word2ix, prefix_words)
     return result
-    # print(''.join(result))
 
 
 # if __name__ == '__main__':
-#     conf={
-#     "max_gen_len":200,  # 生成诗歌最长长度
-#     "prefix_words":'漂亮',  # 不是诗歌的组成部分，用来控制生成诗歌的意境
-#     "start_words" : '雨'  # 诗歌开始
+#     conf={'start_words':"雨",'prefix_words':"天晴", 'max_gen_len':100}
+#     start_words = conf['start_words']    #诗歌开始
+#     prefix_words = conf['prefix_words']  #诗歌语境
+#     max_gen_len = conf['max_gen_len']    #诗歌最大长度
+#     cof={
+#     "max_gen_len": max_gen_len,  # 生成诗歌最长长度
+#     "prefix_words":prefix_words,  # 不是诗歌的组成部分，用来控制生成诗歌的意境
+#     "start_words" : start_words  # 诗歌开始
 #     }
-#     result = gen(**conf)
+#     result = gen(**cof)
 #     result = ''.join(result)
-#     return {'ret1':result}
-#     # print(result)
+#     # return {'ret1':result}
+#     print(result)