Update app.py

app.py

@@ -10,15 +10,8 @@ hf_token = os.getenv("HF_TOKEN")
 # 使用 Hugging Face Token 登录
 login(token=hf_token)
 
-# 加载模型
-pipe = StableDiffusion3Pipeline.from_pretrained("prithivMLmods/SD3.5-Large-Photorealistic-LoRA", 
-                                                revision="main", 
-                                                torch_dtype=torch.float16)
-
-# 如果模型需要LoRA权重文件，可以手动加载
-pipe.load_lora_weights("prithivMLmods/SD3.5-Large-Photorealistic-LoRA", 
-                       weight_name="Photorealistic-SD3.5-Large-LoRA.safetensors")
-
+pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-large", torch_dtype=torch.bfloat16)
+pipe.load_lora_weights("prithivMLmods/SD3.5-Large-Photorealistic-LoRA", weight_name="Photorealistic-SD3.5-Large-LoRA.safetensors")
 pipe.fuse_lora(lora_scale=1.0)
 
 # 定义图像生成函数