Spaces:
Sleeping
Sleeping
no cuda
Browse files
app.py
CHANGED
|
@@ -7,7 +7,6 @@ text_pipeline = pipeline("text-generation", model="isarth/distill_gpt2_story_gen
|
|
| 7 |
|
| 8 |
# 圖像生成模型
|
| 9 |
image_pipeline = DiffusionPipeline.from_pretrained("prompthero/openjourney")
|
| 10 |
-
image_pipeline.to("cuda") # 若部署環境支援 GPU，這樣可加速生成
|
| 11 |
|
| 12 |
# 文字生成函式
|
| 13 |
def generate_story(input_text):
|
|
|
|
| 7 |
|
| 8 |
# 圖像生成模型
|
| 9 |
image_pipeline = DiffusionPipeline.from_pretrained("prompthero/openjourney")
|
|
|
|
| 10 |
|
| 11 |
# 文字生成函式
|
| 12 |
def generate_story(input_text):
|