Update app_2.py

app_2.py

@@ -12,6 +12,8 @@ import datetime
 from pathlib import Path
 from io import BytesIO
 
+
+
 from PIL import Image
 from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline
 from diffusers import AutoencoderKL, UNet2DConditionModel, DDIMScheduler, EulerAncestralDiscreteScheduler, DPMSolverMultistepScheduler
@@ -34,6 +36,7 @@ import cv2
 
 from transformers import AutoModelForImageSegmentation
 from inference_i2mv_sdxl import prepare_pipeline, remove_bg, run_pipeline
+from torchvision import transforms
 
 
 from typing import Optional