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