Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import pipeline
 from gtts import gTTS
 import torch
 import os
-
+from diffusers import DiffusionPipeline
 # ----------------------------
 # 1. 图像描述生成函数
 # ----------------------------
@@ -54,14 +54,9 @@ def load_image_generator():
     device = "cuda" if torch.cuda.is_available() else "cpu"
     
     # 导入 StableDiffusionPipeline
-    from diffusers import StableDiffusionPipeline
-
     # 对于 GPU，采用 fp16 精度以加速推理；否则使用默认精度
     torch_dtype = torch.float16 if device == "cuda" else torch.float32
-    pipe = StableDiffusionPipeline.from_pretrained(
-        "stabilityai/stable-diffusion-v1-5",
-        torch_dtype=torch_dtype
-    )
+    pipe = DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5")
     pipe = pipe.to(device)
     return pipe