Commit
·
6d1cde4
1
Parent(s):
a5284e6
Update app.py
Browse files
app.py
CHANGED
@@ -4,13 +4,13 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
|
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 |
|
12 |
]
|
13 |
|
|
|
|
|
14 |
def text_processing(text):
|
15 |
inputs = [text]
|
16 |
|
|
|
4 |
tokenizer = AutoTokenizer.from_pretrained("google/mt5-small")
|
5 |
model = AutoModelForSeq2SeqLM.from_pretrained("./checkpoint-15000/")
|
6 |
|
|
|
|
|
7 |
examples = [
|
8 |
["猪笼草原产于热带和亚热带地区 现主要分布在东南亚一带 中国广东 广西等地有分布 猪笼草喜欢湿润和温暖半阴的生长环境 不耐寒 怕积水 怕强光 怕干燥 喜欢疏松 肥沃和透气的腐叶土和泥炭土 对光照要求较为严格 猪笼草的繁殖方式包括扦插繁殖 压条繁殖和播种繁殖"],
|
9 |
|
10 |
]
|
11 |
|
12 |
+
inputs=[gr.inputs.Textbox(default=examples[0][0], label="输入文本")]
|
13 |
+
|
14 |
def text_processing(text):
|
15 |
inputs = [text]
|
16 |
|