Update app.py

app.py

@@ -13,7 +13,7 @@ import random
 import time
 from typing import Any, Dict, List, Optional, Union
 from huggingface_hub import hf_hub_download
-from diffusers import DiffusionPipeline, AutoencoderTiny
+from diffusers import DiffusionPipeline, AutoencoderTiny, AutoPipelineForImage2Image
 import safetensors.torch
 from safetensors.torch import load_file
 from pipeline import FluxWithCFGPipeline