Update app.py
Browse files
app.py
CHANGED
|
@@ -8,11 +8,11 @@ import json
|
|
| 8 |
import zipfile
|
| 9 |
from io import BytesIO
|
| 10 |
|
| 11 |
-
# 初始化模型Pipeline(
|
| 12 |
@st.cache_resource
|
| 13 |
def load_models():
|
| 14 |
# 剧本生成(T5模型)
|
| 15 |
-
script_pipe = pipeline("text2text-generation", model="mrm8488/t5-base-finetuned-
|
| 16 |
|
| 17 |
# 分镜生成(BART模型)
|
| 18 |
storyboard_pipe = pipeline("text-generation", model="philschmid/bart-large-cnn-samsum")
|
|
|
|
| 8 |
import zipfile
|
| 9 |
from io import BytesIO
|
| 10 |
|
| 11 |
+
# 初始化模型Pipeline(使用修正后的模型名称)
|
| 12 |
@st.cache_resource
|
| 13 |
def load_models():
|
| 14 |
# 剧本生成(T5模型)
|
| 15 |
+
script_pipe = pipeline("text2text-generation", model="mrm8488/t5-base-finetuned-common_gen")
|
| 16 |
|
| 17 |
# 分镜生成(BART模型)
|
| 18 |
storyboard_pipe = pipeline("text-generation", model="philschmid/bart-large-cnn-samsum")
|