Update app.py

app.py

@@ -6,12 +6,11 @@ import gradio as gr
 import numpy as np
 import torch
 from PIL import Image
-from diffusers import DiffusionPipeline, StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler, StableDiffusionXLInstructPix2PixPipeline, AutoencoderKL
+from diffusers import DiffusionPipeline, StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler, StableDiffusionInstructPix2PixPipeline, AutoencoderKL
 from custom_pipeline import CosStableDiffusionXLInstructPix2PixPipeline
 from huggingface_hub import hf_hub_download
 from huggingface_hub import InferenceClient
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.float16
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 
@@ -34,8 +33,7 @@ To optimize image results:
 """ 
 
 # Image Editor
-pipe_edit = StableDiffusionXLInstructPix2PixPipeline.from_pretrained( "vinesmsuic/magicbrush-jul7", torch_dtype=torch.float16, vae=vae ).to("cuda")
-
+pipe_edit = StableDiffusionInstructPix2PixPipeline.from_pretrained( "vinesmsuic/magicbrush-jul7", torch_dtype=torch.float16 ).to("cuda")
 
 # Generator
 @spaces.GPU(duration=30, queue=False)