Jellyfish042 commited on
Commit
a5284e6
·
1 Parent(s): 9c6a874

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -2
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(fn = text_processing, inputs='text', outputs='text', title='Punctuation Mark Prediction', description='本模型主要用于语言识别模型输出的后处理。\n输入无符号句子,需要打标点处用空格隔开,返回带标点句子。\n仅支持中文,因为训练数据中只有中文。', examples=examples)
 
 
 
 
 
 
 
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)