Update app.py

app.py

@@ -1,13 +1,13 @@
 import gradio as gr
 import torch
-from diffusers import AutoPipelineForImage2Image
+from diffusers import StableDiffusionImg2ImgPipeline
 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"
 
-pipe = AutoPipelineForImage2Image.from_pretrained(
+pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
     model_id,
     torch_dtype=torch.float16 if device == "cuda" else torch.float32,
     safety_checker=None