diff --git a/main.py b/main.py
index b6be27a..ac4e9cd 100644
--- a/main.py
+++ b/main.py
@@ -23,7 +23,7 @@
     batch_size = 128
     maxlen = 125  # 超过这个长度的之后字被丢弃，小于这个长度的在前面补空格
     plot_every = 20  # 每20个batch 可视化一次
-    # use_env = True # 是否使用visodm
+    use_env = True # 是否使用visodm
     env = 'poetry'  # visdom env
     max_gen_len = 200  # 生成诗歌最长长度
     debug_file = '/tmp/debugp'
@@ -189,11 +189,13 @@
 
 
 def gen(**kwargs):
+# def gen(opt):
     """
     提供命令行接口，用以生成相应的诗
     """
 
     for k, v in kwargs.items():
+        print(k,v)
         setattr(opt, k, v)
     data, word2ix, ix2word = get_data(opt)
     model = PoetryModel(len(word2ix), 128, 256);
@@ -228,7 +230,11 @@
 
 
 if __name__ == '__main__':
-    # import fire
-    #
-    # fire.Fire()
-    gen()
+    conf={
+    # "max_gen_len":200,  # 生成诗歌最长长度
+    # "prefix_words":'漂亮',  # 不是诗歌的组成部分，用来控制生成诗歌的意境
+    # "start_words" : '雨'  # 诗歌开始
+    }
+    result = gen(**conf)
+    result = ''.join(result)
+    print(result)