Update app.py

app.py

@@ -3,15 +3,18 @@ import torch
 from diffusers import StableDiffusion3Pipeline
 import os
 from huggingface_hub import login
+
+# 获取Hugging Face Token
 hf_token = os.environ.get("HF_TOKEN")
 login(token=hf_token)
 
-
 # 加载模型并配置
 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")
+
+# 如果有GPU，转移到GPU
+pipe.to("cuda")
 
 # 定义图像生成函数，添加种子参数
 def generate_image(prompt, seed):
@@ -42,7 +45,7 @@ with gr.Blocks() as demo:
 
     # 按钮触发事件
     generate_btn = gr.Button("Generate Image")
-    generate_btn.click(generate_image, inputs=[prompt_input, seed_input], outputs=output_image)
+    generate_btn.click(fn=generate_image, inputs=[prompt_input, seed_input], outputs=output_image)
 
 # 启动Gradio应用
 demo.launch()