Update app.py

app.py

@@ -19,7 +19,7 @@ if not os.path.exists(LORA_DIR):
 
 # 2. 训练完成后，加载基座模型 + LoRA
 print("🔄 正在加载模型和 LoRA ...")
-pipe = StableDiffusionPipeline.from_pretrained("silveroxides/Chroma-GGUF", torch_dtype=torch.float16).to("cuda")
+pipe = StableDiffusionPipeline.from_pretrained("prompthero/openjourney", torch_dtype=torch.float16).to("cuda")
 pipe.unet = PeftModel.from_pretrained(pipe.unet, LORA_DIR)
 print("✅ 模型加载完成，可以开始生成图片！")