Update app.py

app.py

@@ -1,17 +1,19 @@
 import gradio as gr
 import torch
-from diffusers import StableDiffusionImg2ImgPipeline
+from diffusers import AutoPipelineForImage2Image
 from PIL import Image
 from typing import List
 #StableDiffusionImg2ImgPipeline
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_id = "stabilityai/stable-diffusion-2-1-unclip-small"
+model_id = "black-forest-labs/FLUX.1-dev"
 
-pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
+pipe = AutoPipelineForImage2Image.from_pretrained(
     model_id,
     torch_dtype=torch.float16 if device == "cuda" else torch.float32,
     safety_checker=None
 ).to(device)
+
+pipe.load_lora_weights("openfree/flux-chatgpt-ghibli-lora")
 pipe.enable_attention_slicing()
 
 styles = {