[update]edit main

main.py

@@ -37,7 +37,7 @@ def get_args():
 description = """
 ## Qwen-7B
 
-基于 [Qwen-7B](https://huggingface.co/qgyd2021/Qwen-7B) 模型, 在 [chinese_modern_poetry](https://huggingface.co/datasets/Iess/chinese_modern_poetry) 的 prompt 数据集上训练了 2 个 epoch. 
+基于 [Qwen-7B](https://huggingface.co/qgyd2021/Qwen-7B) 模型, 在 [chinese_modern_poetry](https://huggingface.co/datasets/Iess/chinese_modern_poetry) 数据集上训练了 2 个 epoch. 
 
 可用于生成现代诗. 如下: 
 使用下列意象写一首现代诗：智慧，刀刃. 
@@ -46,7 +46,10 @@ description = """
 
 examples = [
     "使用下列意象写一首现代诗：石头，森林",
-    "使用下列意象写一首现代诗：花，纱布"
+    "使用下列意象写一首现代诗：花，纱布",
+    "使用下列意象写一首现代诗：山壁，彩虹，诗句，山坡，泪",
+    "使用下列意象写一首现代诗：味道，黄金，名字，银子，女人",
+    "使用下列意象写一首现代诗：乳房，触感，车速，星星，路灯"
 
 ]
 
@@ -100,6 +103,8 @@ def main():
         return [(text, response)]
 
     def fn_stream(text: str):
+        text = str(text).strip()
+
         input_ids = tokenizer(
             text,
             return_tensors="pt",
@@ -127,7 +132,8 @@ def main():
 
         output = ""
         for output_ in streamer:
-            output_ = output_.strip().replace(tokenizer.eos_token, "").strip()
+            output_ = output_.replace(text, "")
+            output_ = output_.replace(tokenizer.eos_token, "")
 
             output += output_