Update

app.py

@@ -4,7 +4,9 @@ from enum import Enum
 
 
 st.header("WeLM Demo 初体验")
-
+st.text('Tips: ')
+st.text("* WeLM 不是一个直接的对话机器人，而是一个补全用户输入信息的生成模型")
+st.text("* 修改Prompts请先参考 https://welm.weixin.qq.com/docs/introduction/")
 
 class Task(str, Enum):
     DIALOG_JOURNAL = "对话-采访"
@@ -50,8 +52,7 @@ Elon Musk：""",
 改写这段话来描述小丑。{男人站在马戏团外一边拿着气球一边打电话}
 有这样一段文本，{风铃声响起}
 改写这段话写的更加丰富。{我对这个风铃的感情是由它的铃声引起的。每当风吹来时，风铃发出非常动听的声音，听起来是那么乐观、豁达，像一个小女孩格格的笑声。}
-有这样一段文本，{我想家了}
-改写这段话包含更多悲伤的感情。{""",
+""",
     Task.COMPLETION: """“八月十八潮，壮观天下无。”这是北宋大诗人苏东坡咏赞钱塘秋潮的千古名句。千百年来，钱塘江以其奇特卓绝的江潮，不知倾倒了多少游人看客。
 每年的农历八月十八前后，是观潮的最佳时节。这期间，秋阳朗照，金风宜人，钱塘江口的海塘上，游客群集，兴致盎然，争睹奇景。
 """,
@@ -59,7 +60,13 @@ Elon Musk：""",
 }
 
 prompt = task2prompt[Task[task_type]]
-all_input = st.text_area('模型输入', value=prompt, height=400)
+if task_type == 'REWRITE':
+    pre_input = st.text_area('输入样例', value=prompt, height=400)
+    rewrite_text = "有这样一段文本，{我想家了}\n改写这段话包含更多悲伤的感情。{"
+    rewrite_text = st.text_area('你的输入', value=rewrite_text, height=100)
+    all_input = prompt + rewrite_text  
+else:
+    all_input = st.text_area('模型输入', value=prompt, height=400)
 all_input = all_input.rstrip('\\n')