Leo Liu commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,20 +11,6 @@ def img2text(url):
|
|
| 11 |
|
| 12 |
# text2story
|
| 13 |
def text2story(text):
|
| 14 |
-
# 结构化prompt模板（强制模型遵循起承转合）
|
| 15 |
-
prompt = f"""Write a COMPLETE fairy tale for children aged 3-10. Follow these rules:
|
| 16 |
-
[STORY STRUCTURE]
|
| 17 |
-
1. BEGINNING: Introduce characters and setting (20-30 words)
|
| 18 |
-
2. MIDDLE: Describe a magical problem (40-60 words)
|
| 19 |
-
3. ENDING: Show how kindness/courage solves it (30-40 words)
|
| 20 |
-
|
| 21 |
-
[EXAMPLE]
|
| 22 |
-
Input: "a forest with glowing mushrooms"
|
| 23 |
-
Output: "Little fox Luna found glowing mushrooms deep in the forest. They were fairy homes! When a storm threatened to destroy them, Luna bravely built a leaf umbrella to protect the tiny homes. The grateful fairies made her forest guardian, and they danced under moonbeams every night."
|
| 24 |
-
|
| 25 |
-
[INPUT SCENE]: {text}
|
| 26 |
-
[STORY]:""" # 用方括号明确区分指令和内容
|
| 27 |
-
|
| 28 |
pipe = pipeline(
|
| 29 |
"text-generation",
|
| 30 |
model="pranavpsv/genre-story-generator-v2",
|
|
@@ -40,7 +26,7 @@ Output: "Little fox Luna found glowing mushrooms deep in the forest. They were f
|
|
| 40 |
)
|
| 41 |
|
| 42 |
# 生成并后处理
|
| 43 |
-
raw_output = pipe(
|
| 44 |
|
| 45 |
# 提取核心内容（适配模型输出模式）
|
| 46 |
if "[STORY]:" in raw_output:
|
|
|
|
| 11 |
|
| 12 |
# text2story
|
| 13 |
def text2story(text):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
pipe = pipeline(
|
| 15 |
"text-generation",
|
| 16 |
model="pranavpsv/genre-story-generator-v2",
|
|
|
|
| 26 |
)
|
| 27 |
|
| 28 |
# 生成并后处理
|
| 29 |
+
raw_output = pipe(text, return_full_text=False)[0]['generated_text']
|
| 30 |
|
| 31 |
# 提取核心内容（适配模型输出模式）
|
| 32 |
if "[STORY]:" in raw_output:
|