Update app.py

app.py

@@ -11,19 +11,21 @@ from diffusers import StableDiffusionPipeline
 
 
 
-# 1. 选择一个基础模型，例如 SD 1.5
-base_model_id = "runwayml/stable-diffusion-v1-5"
+# # 1. 选择一个基础模型，例如 SD 1.5
+# base_model_id = "runwayml/stable-diffusion-v1-5"
 
-# 2. 加载基础模型
-pipe = StableDiffusionPipeline.from_pretrained(
-    base_model_id,
-    torch_dtype=torch.float32
-)
+# # 2. 加载基础模型
+# pipe = StableDiffusionPipeline.from_pretrained(
+#     base_model_id,
+#     torch_dtype=torch.float32
+# )
 
-# 3. 加载 LoRA 权重
-lora_model_id = "openfree/flux-chatgpt-ghibli-lora"
-pipe.load_lora_weights(lora_model_id)
+# # 3. 加载 LoRA 权重
+# lora_model_id = "openfree/flux-chatgpt-ghibli-lora"
+# pipe.load_lora_weights(lora_model_id)
 
+pipe = StableDiffusionImg2ImgPipeline.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.bfloat16)
+pipe.load_lora_weights('openfree/flux-chatgpt-ghibli-lora', weight_name='flux-chatgpt-ghibli-lora.safetensors')
 
 # Move pipeline to GPU if available
 device = "cuda" if torch.cuda.is_available() else "cpu"