Update app.py

app.py

@@ -1,13 +1,13 @@
 import gradio as gr
 import torch
-from diffusers import StableDiffusionImg2ImgPipeline
+from diffusers import FluxImg2ImgPipeline
 from PIL import Image
 from typing import List
-
+#StableDiffusionImg2ImgPipeline
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_id = "nitrosocke/Ghibli-Diffusion"
+model_id = "openfree/flux-chatgpt-ghibli-lora"
 
-pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
+pipe = FluxImg2ImgPipeline.from_pretrained(
     model_id,
     torch_dtype=torch.float16 if device == "cuda" else torch.float32,
     safety_checker=None