Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from diffusers import StableDiffusion3Pipeline
 import torch
-
+import os
 from huggingface_hub import login
 
 # 通过环境变量获取 Token
@@ -11,7 +11,15 @@ hf_token = os.getenv("HF_TOKEN")
 login(token=hf_token)
 
 # 加载模型
-pipe = StableDiffusion3Pipeline.from_pretrained("prithivMLmods/SD3.5-Large-Photorealistic-LoRA", torch_dtype=torch.float16)
+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.fuse_lora(lora_scale=1.0)
 
 # 定义图像生成函数
 def generate_image(prompt):