Update app.py

app.py

@@ -7,7 +7,7 @@ import numpy as np
 from PIL import Image
 import spaces
 import torch
-from diffusers import StableDiffusion3Pipeline, DPMSolverMultistepScheduler, AutoencoderKL, AutoPipelineForImage2Image
+from diffusers import StableDiffusion3Pipeline,StableDiffusion3Img2ImgPipeline, DPMSolverMultistepScheduler, AutoencoderKL
 from huggingface_hub import snapshot_download
 
 huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
@@ -34,7 +34,7 @@ ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 pipe = StableDiffusion3Pipeline.from_pretrained(model_path, torch_dtype=torch.float16)
-img2img_pipe = AutoPipelineForImage2Image.from_pretrained(model_path, torch_dtype=torch.float16)
+img2img_pipe = StableDiffusion3Img2ImgPipeline.from_pretrained(model_path, torch_dtype=torch.float16)
 
 def save_image(img):
     unique_name = str(uuid.uuid4()) + ".png"