Update app.py
Browse files
app.py
CHANGED
|
@@ -25,7 +25,7 @@ def generate_story(caption):
|
|
| 25 |
# 使用 GPU 进行文本生成操作
|
| 26 |
story_generator = pipeline(
|
| 27 |
"text-generation",
|
| 28 |
-
model="
|
| 29 |
device=device_id
|
| 30 |
)
|
| 31 |
messages = (
|
|
@@ -35,7 +35,7 @@ def generate_story(caption):
|
|
| 35 |
result = story_generator(messages, max_length=300, num_return_sequences=1)
|
| 36 |
story = result[0]['generated_text']
|
| 37 |
return story
|
| 38 |
-
|
| 39 |
# 以下部分为生成插图示例代码，已注释。如果需要使用 GPU，请取消注释并确保 diffusers 相关依赖已经安装
|
| 40 |
# @st.cache_resource
|
| 41 |
# def load_image_generator():
|
|
|
|
| 25 |
# 使用 GPU 进行文本生成操作
|
| 26 |
story_generator = pipeline(
|
| 27 |
"text-generation",
|
| 28 |
+
model="openai-community/gpt2-medium",
|
| 29 |
device=device_id
|
| 30 |
)
|
| 31 |
messages = (
|
|
|
|
| 35 |
result = story_generator(messages, max_length=300, num_return_sequences=1)
|
| 36 |
story = result[0]['generated_text']
|
| 37 |
return story
|
| 38 |
+
000000000
|
| 39 |
# 以下部分为生成插图示例代码，已注释。如果需要使用 GPU，请取消注释并确保 diffusers 相关依赖已经安装
|
| 40 |
# @st.cache_resource
|
| 41 |
# def load_image_generator():
|