ake178178 commited on
Commit
0ce1c76
·
verified ·
1 Parent(s): 2c5757f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ image_to_text_pipe = pipeline("image-to-text",
8
 
9
  # 加载第二个pipeline,用于根据文本生成故事
10
  text_to_story_pipe = pipeline("text-generation",
11
- model="meta-llama/Llama-2-7b-hf")
12
 
13
  # 设置随机种子以保证结果的一致性
14
  set_seed(42)
 
8
 
9
  # 加载第二个pipeline,用于根据文本生成故事
10
  text_to_story_pipe = pipeline("text-generation",
11
+ model="mistralai/Mistral-7B-Instruct-v0.2")
12
 
13
  # 设置随机种子以保证结果的一致性
14
  set_seed(42)