Update app.py

app.py

@@ -3,12 +3,12 @@ import modin.pandas as pd
 import torch
 import numpy as np
 from PIL import Image
-from diffusers import AutoPipelineForImage2Image
+from diffusers import ShapEImg2ImgPipeline
 from diffusers.utils import load_image
 import math
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = AutoPipelineForImage2Image.from_pretrained("openai/shap-e-img2img", torch_dtype=torch.float16) if torch.cuda.is_available() else AutoPipelineForImage2Image.from_pretrained("openai/shap-e-img2img")
+pipe = ShapEImg2ImgPipeline.from_pretrained("openai/shap-e-img2img", torch_dtype=torch.float16) if torch.cuda.is_available() else ShapEImg2ImgPipeline.from_pretrained("openai/shap-e-img2img")
 pipe = pipe.to(device)
 
 def resize(value,img):