diff --git a/handler.py b/handler.py
index f78ee8e..fba7980 100644
--- a/handler.py
+++ b/handler.py
@@ -1,3 +1,5 @@
+import main
+
 def handle(conf):
     """
     该方法是部署之后，其他人调用你的服务时候的处理方法。
@@ -8,16 +10,13 @@
     参数请放到params字典中，我们会自动解析该变量。
     """
 
-    if __name__ == '__main__':
-        import main
-        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)
-        return {'ret1':result}
-    
\ No newline at end of file
+    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)
+    return {'result':result}
diff --git a/main.py b/main.py
index 75d10df..6c3b9bb 100644
--- a/main.py
+++ b/main.py
@@ -227,17 +227,16 @@
     return result
 
 
-# if __name__ == '__main__':
-#     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(**cof)
-#     result = ''.join(result)
-#     # return {'ret1':result}
-#     print(result)
+if __name__ == '__main__':
+    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(**cof)
+    # return {'ret1':result}
+    print(result)