Update app.py

app.py

@@ -7,10 +7,10 @@ import torch
 from PIL import Image
 from diffusers import StableDiffusionPipeline
 # Load the model
-model_id = "nitrosocke/Ghibli-Diffusion"
-# model_id = "openfree/flux-chatgpt-ghibli-lora"
-pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
-
+# model_id = "nitrosocke/Ghibli-Diffusion"
+# pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
+model_id = "black-forest-labs/FLUX.1-dev"
+pipe =DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16)
 
 
 # # 1. 选择一个基础模型，例如 SD 1.5