Jessiesj commited on
Commit
9e40f8f
·
verified ·
1 Parent(s): 820b6c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,11 +8,11 @@ import json
8
  import zipfile
9
  from io import BytesIO
10
 
11
- # 初始化模型Pipeline(全部使用Hugging Face预训练模型)
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")
 
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")