Update app.py

app.py

@@ -2,9 +2,11 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 tokenizer = AutoTokenizer.from_pretrained("google/mt5-small")
-model = AutoModelForSeq2SeqLM.from_pretrained("./checkpoint-15000/")
+model = AutoModelForSeq2SeqLM.from_pretrained("./model/")
 
 def text_processing(text):
+    
+    text = text + ' ' if text[-2:] != ' ' else text  # 在末尾加上空格有利于模型预测
     inputs = [text]
 
     # Tokenize and prepare the inputs for model
@@ -20,7 +22,7 @@ def text_processing(text):
     return decoded_output[0]
 
 examples = [
-    ["猪笼草原产于热带和亚热带地区 现主要分布在东南亚一带 中国广东 广西等地有分布 猪笼草喜欢湿润和温暖半阴的生长环境 不耐寒 怕积水 怕强光 怕干燥 喜欢疏松 肥沃和透气的腐叶土和泥炭土 对光照要求较为严格 猪笼草的繁殖方式包括扦插繁殖 压条繁殖和播种繁殖"],
+    ["我们的价值观是 富强 民主 文明 和谐"],
     ["都什么年代了 还在抽传统香烟"]
 ]