Update app.py

app.py

@@ -1,14 +1,26 @@
 import gradio as gr
+import torch
+from diffusers import StableDiffusion3Pipeline
 
 # 加载模型并配置
-model = gr.load("models/prithivMLmods/SD3.5-Large-Photorealistic-LoRA")
-# .launch()
+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)
+# pipe.to("cuda")
 
-# 定义图像生成函数，添加seed参数
+# 定义图像生成函数，添加种子参数
 def generate_image(prompt, seed):
-    # 调用模型的predict方法并传递种子
-    result = model.predict(prompt=prompt, seed=seed)
-    return result
+    # 设置种子
+    generator = torch.manual_seed(seed)
+    
+    # 使用模型生成图像
+    image = pipe(prompt=prompt,
+                 num_inference_steps=24, 
+                 guidance_scale=4.0,
+                 width=960, height=1280,
+                 generator=generator).images[0]
+    
+    return image
 
 # 创建Gradio界面
 with gr.Blocks() as demo: