diff --git a/main_scokt.py b/main_scokt.py
index 1ac0300..011ae6d 100644
--- a/main_scokt.py
+++ b/main_scokt.py
@@ -209,6 +209,11 @@ def main(question, title, top):
 
 
 def model_generate_stream(prompt):
+    '''
+    流式问答，可切换成非流式的
+    :param prompt:
+    :return:
+    '''
     messages = [
         {"role": "user", "content": prompt}
     ]