Commit
·
a5284e6
1
Parent(s):
9c6a874
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,8 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
|
4 |
tokenizer = AutoTokenizer.from_pretrained("google/mt5-small")
|
5 |
model = AutoModelForSeq2SeqLM.from_pretrained("./checkpoint-15000/")
|
6 |
|
7 |
-
|
|
|
8 |
examples = [
|
9 |
["猪笼草原产于热带和亚热带地区 现主要分布在东南亚一带 中国广东 广西等地有分布 猪笼草喜欢湿润和温暖半阴的生长环境 不耐寒 怕积水 怕强光 怕干燥 喜欢疏松 肥沃和透气的腐叶土和泥炭土 对光照要求较为严格 猪笼草的繁殖方式包括扦插繁殖 压条繁殖和播种繁殖"],
|
10 |
|
@@ -25,6 +26,13 @@ def text_processing(text):
|
|
25 |
|
26 |
return decoded_output[0]
|
27 |
|
28 |
-
iface = gr.Interface(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
29 |
|
30 |
iface.launch(inline=False)
|
|
|
4 |
tokenizer = AutoTokenizer.from_pretrained("google/mt5-small")
|
5 |
model = AutoModelForSeq2SeqLM.from_pretrained("./checkpoint-15000/")
|
6 |
|
7 |
+
inputs=[gr.inputs.Textbox(default=examples[0][0], label="输入文本")]
|
8 |
+
|
9 |
examples = [
|
10 |
["猪笼草原产于热带和亚热带地区 现主要分布在东南亚一带 中国广东 广西等地有分布 猪笼草喜欢湿润和温暖半阴的生长环境 不耐寒 怕积水 怕强光 怕干燥 喜欢疏松 肥沃和透气的腐叶土和泥炭土 对光照要求较为严格 猪笼草的繁殖方式包括扦插繁殖 压条繁殖和播种繁殖"],
|
11 |
|
|
|
26 |
|
27 |
return decoded_output[0]
|
28 |
|
29 |
+
iface = gr.Interface(
|
30 |
+
fn=text_processing,
|
31 |
+
inputs=[gr.inputs.Textbox(default=examples[0][0], label="输入文本")],
|
32 |
+
outputs='text',
|
33 |
+
title='Punctuation Mark Prediction',
|
34 |
+
description='本模型主要用于语言识别模型输出的后处理。\n输入无符号句子,需要打标点处用空格隔开,返回带标点句子。\n仅支持中文,因为训练数据中只有中文。',
|
35 |
+
examples=examples
|
36 |
+
)
|
37 |
|
38 |
iface.launch(inline=False)
|